hdu1520 Anniversary party（最大独立集 树形dp）

题目链接：点击打开链接

题目描述：现有一棵树，树上每个结点都有一个权值，问从中选一些点，这些点两两之间不直接连接，问权值最大为多少？

解题思路：很裸的一道树上最大独立集问题 树形dp即可

dp[i][0]:不选i节点 dp[i][0]+=max(dp[t][0],dp[t][1]);

dp[i][1]:选i节点     dp[i][1]+=dp[t][0];

代码：

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <iostream>
#define MAXN 6010
using namespace std;
int head[MAXN],tol;
struct Edge{
    int v,next;
}edge[MAXN];
void addEdge(int u,int v){
    edge[tol].v=v;edge[tol].next=head[u];head[u]=tol++;
}
int max(int a,int b){
    return a>b?a:b;
}
int dp[MAXN][2];
int n,value[MAXN];
bool vis[MAXN];
void DP(int u,int p){
    dp[u][0]=0;
    dp[u][1]=value[u];
    int k,to;
    for(k=head[u];k!=-1;k=edge[k].next){
        to=edge[k].v;
        if(to==p) continue;
        DP(to,u);
        dp[u][0]+=max(dp[to][0],dp[to][1]);
        dp[u][1]+=dp[to][0];
    }
}
int main(){
    while(scanf("%d",&n)!=EOF){
        for(int i=1;i<=n;++i) scanf("%d",&value[i]);
        int u,v;
        tol=0;memset(head,-1,sizeof(head));
        memset(vis,false,sizeof(vis));
        while(scanf("%d%d",&u,&v)!=EOF&&(u!=0||v!=0)) {addEdge(v,u);vis[u]=true;}
        int root;
        for(int i=1;i<=n;i++){if(!vis[i]) {root=i;break;}}
        DP(root,root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}