hdoj1520（入门树形dp）

题目链接：https://vjudge.net/problem/HDU-1520

题意：和luogu那道没有上司的舞会一样的题，给定一棵带点权的树，父结点和子结点不能同时选，问怎么选使得权值和最大，求最大值即可。

思路：最近开始肝树形dp，从入门题开始QAQ，加油！

　　　用dp[u][0]表示结点u不选，dp[u][1]表示选，vi是结点u的子结点，那么：

　　　　dp[u][0]=sum(max(dp[vi][0],dp[vi][1]))

　　　　dp[u][1]=val[u]+sum(dp[vi][0])

　　　一次dfs就ok了，结果为max(dp[root][0],dp[root][1])，root要自己找。

AC代码：

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn=6005;
int n,cnt,head[maxn],val[maxn],indeg[maxn],dp[maxn][2];
struct node{
    int v,nex;
}edge[maxn];

void adde(int u,int v){
    edge[++cnt].v=v;
    edge[cnt].nex=head[u];
    head[u]=cnt;
}

void dfs(int u){
    dp[u][1]=val[u];
    for(int i=head[u];i;i=edge[i].nex){
        int v=edge[i].v;
        dfs(v);
        dp[u][0]+=max(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][0];
    }
}

int main(){
    while(~scanf("%d",&n)){
        cnt=0;
        for(int i=1;i<=n;++i){
            scanf("%d",&val[i]);
            head[i]=0,indeg[i]=0,dp[i][0]=0;
        }
        int a,b;
        while(scanf("%d%d",&a,&b),a&&b){
            indeg[a]=1;
            adde(b,a);
        }
        int root;
        for(int i=1;i<=n;++i)
            if(!indeg[i]){
                root=i;
                break;
            }
        dfs(root);
        printf("%d
",max(dp[root][0],dp[root][1]));
    }
    return 0;
}