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 = 1001;
const int inf = 0x3f3f3f3f;
int n, cnt;
int dp[N][2];
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] = 1;
    dp[u][0] = 0;  dp[u][1] = 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][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(){
    int i, j, ans;
    scanf("%d", &n);
    CLR(head, -1);  CLR(dp, 0);  CLR(vis, 0);
    cnt = 0;
    for(i = 1; i <= n; ++i){
        scanf("%d%d%d", &p[i].x, &p[i].y, &p[i].c);
        for(j = 1; j < i; ++j){
            if(abs(p[i].x - p[j].x) + abs(p[i].y - p[j].y) == 1){
                add(i, j); add(j, i);
            }
        }
    }
    dfs(1);
    ans = max(dp[1][0], dp[1][1]);
    printf("%d
", ans);
    return 0;
}