POJ-1192-最優連通子集
阿新 • • 發佈:2019-02-04
本來書上是放在最短路徑那章的,寫了半天發現很麻煩,一搜索才知道這個題應該用樹形DP做,
算是比較簡單的樹形DP題吧。
程式碼:
#include<cstdio> #include<cstring> #include<iostream> #include<cstdlib> #include<queue> using namespace std; const int maxn=1001; const int inf=1<<29; int n,py[maxn],px[maxn],c[maxn],dp[maxn][2]; int e,pnt[maxn*maxn],nxt[maxn*maxn],head[maxn]; void addedge(int u,int v) { pnt[e]=v;nxt[e]=head[u];head[u]=e++; } void DFS(int u,int f) { dp[u][0]=0; dp[u][1]=c[u]; for(int i=head[u];i!=-1;i=nxt[i]) { int v=pnt[i]; if(v!=f) { DFS(v,u); dp[u][0]=max(dp[u][0],max(dp[v][0],dp[v][1])); if(dp[v][1]>0) dp[u][1]+=dp[v][1]; } } } int main() { while(~scanf("%d",&n)) { e=0; memset(head,-1,sizeof(head)); memset(dp,0,sizeof(dp)); for(int i=1;i<=n;i++) { scanf("%d%d%d",&px[i],&py[i],&c[i]); for(int j=1;j<i;j++) if(abs(px[i]-px[j])+abs(py[i]-py[j])==1) { addedge(j,i); addedge(i,j); } } DFS(1,-1); printf("%d\n",max(dp[1][0],dp[1][1])); } return 0; }