poj1192 最优连通子集（树形dp）

题目链接：poj1192 最优连通子集

求一棵无向树的最大子树和。。类似于求最大子段和的办法，树形dp。

dp[i][0]：以i为根，不包括 i 结点的子树最大权

dp[i][1]：以i为根，包括 i 结点的子树的最大权

dp[i][0] = max(dp[j][0] , dp[j][1])   （j 为 i 的儿子）

dp[i][1] +=  dp[j][1] （dp[j][1] > 0，j 为 i 的儿子）

 #include<cstdio>
 #include<cstring>
 #include<cmath>
 #include<algorithm>
 #define CLR(a,b) memset((a),(b),sizeof((a)))
 using namespace std;
 const int N = ;
 const int inf = 0x3f3f3f3f;
 int n, cnt;
 int dp[N][];
 int head[N];
 bool vis[N];
 struct edge{
     int nex, to;
 }e[N*N];
 struct point{
     int x, y, c;
 }p[N];
 void add(int u, int v){
     e[cnt].to = v;
     e[cnt].nex = head[u];
     head[u] = cnt++;
 }
 void dfs(int u){
     vis[u] = ;
     dp[u][] = ;  dp[u][] = p[u].c;
     for(int i = head[u]; ~i; i = e[i].nex){
         int v = e[i].to;
         if(!vis[v]){
             dfs(v);
             dp[u][] = max(dp[u][], max(dp[v][], dp[v][]));
             if(dp[v][] > )
                 dp[u][] += dp[v][];
         }
     }
 }
 int main(){
     int i, j, ans;
     scanf("%d", &n);
     CLR(head, -);  CLR(dp, );  CLR(vis, );
     cnt = ;
     for(i = ; i <= n; ++i){
         scanf("%d%d%d", &p[i].x, &p[i].y, &p[i].c);
         for(j = ; j < i; ++j){
             if(abs(p[i].x - p[j].x) + abs(p[i].y - p[j].y) == ){
                 add(i, j); add(j, i);
             }
         }
     }
     dfs();
     ans = max(dp[][], dp[][]);
     printf("%d\n", ans);
     return ;
 }