题意
某大学有N个职员,编号为1~N。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
解析
经典题,dp[i][0]表示不选i时能取得的最大利益,dp[i][1]相反,状态转移方程就是dp[i][0] += max(dp[j][0],dp[j][1]),j为i的儿子,dp[i][1] += dp[j][0]。
#include <cstdio>
#include <algorithm>
#define Rep( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i<=(i##_END);i++)
#define For( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i++)
#define Lop( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i>=(i##_END);i--)
#define Dnt( i , _begin , _end ) for(int i=(_begin),i##_END=(_end);i!=(i##_END);i--)
using std :: max;
using std :: min;
const int maxx = 50000 + 25;
int n,m,x,y,z,num,ans,rt;
int f[maxx][2];
int a[maxx],ftr[maxx];
int to[maxx<<1],nxt[maxx<<1],head[maxx];
void Ins(int x,int y){
to[++num] = y;nxt[num] = head[x];head[x] = num;
}
void Dfs(int x,int pre){
f[x][1] = a[x];
for(int i=head[x];i;i=nxt[i]){
int now = to[i];
if(now != pre){
Dfs(now,x);
f[x][0] += max(f[now][1],f[now][0]);
f[x][1] += f[now][0];
}
}
}
int main(){
scanf("%d",&n);
Rep( i , 1 , n ) scanf("%d",&a[i]);
while(1){
scanf("%d%d",&x,&y);
if((!x) && (!y)) break;
Ins(x,y);Ins(y,x);
ftr[x] = y;
}
Rep( i , 1 , n ) if(!ftr[i]) rt = i;
Dfs(rt,0);
ans = max(f[rt][0],f[rt][1]);
printf("%d\n",ans);
return 0;
}