Hdu1520-Anniversary party （树形dp）（入门题）

传送门：Anniversary party

题意：公司内的员工关系表是一棵树，每个员工有一个权值，在举行年会时一个员工与他的直属上司不能同时出现（即直接相邻的节点不能同时选取），求最大权值。

思路：树形dp入门题。

选取节点 1 为根，对于每个节点 u 和它的子节点 v：

如果选取节点 u，则节点 v 必不能被选：

dp[u][1] += dp[v][0]

如果不选取节点 u，则选择节点 v 贡献大的一个（题目的要求是不能同时选取，当然可以同时不选取）：

dp[u][0] += max(dp[v][0], dp[v][1])

tips：题目还有一个坑点是多组数据没有指明。

 

AC代码：

#include 
#include 
#include 
using namespace std;
const int maxn=6e3+7;
struct node{
    int to,nxt;
}e[maxn*2];
int a[maxn],tot,head[maxn],dp[maxn][2];
void add(int u,int v){
    e[tot].nxt=head[u];
    e[tot].to=v;
    head[u]=tot++;
    return;
}
void add_edge(int u,int v){
    add(u,v);
    add(v,u);
    return;
}
void dfs(int rt,int p){
    for(int i=head[rt];~i;i=e[i].nxt){
        int to=e[i].to;
        if(to==p) continue;
        dfs(to,rt);
        dp[rt][1]+=dp[to][0];
        dp[rt][0]+=max(dp[to][0],dp[to][1]);
    }
    dp[rt][1]+=a[rt];
    return;
}
int main(void){
    int n;
    while(~scanf("%d",&n)){
        tot=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            dp[i][0]=dp[i][1]=0;
            head[i]=-1;
        }
        int u,v;while(~scanf("%d%d",&u,&v)){
            if(!u&&!v) break;
            add_edge(u,v);
        }
        dfs(1,0);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
    return 0;
}