描述
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
输入格式
第一行一个整数N。(1<=N<=3000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。
输出格式
输出最大的快乐指数。
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;int n,r[3010],d[3010][2],ans=0;
bool edge[3010][3010],vis[3010];int dp(int a){d[a][1]=r[a];d[a][0]=0;for(int i=1;i<=n;i++){if(edge[a][i]){if(!vis[i]) dp(i);d[a][1]+=d[i][0];d[a][0]+=max(d[i][1],d[i][0]); }}vis[a]=true;return max(d[a][1],d[a][0]);
}int main(){memset(edge,0,sizeof(edge));scanf("%d",&n);for(int i=1;i<=n;i++) scanf("%d",&r[i]);for(int i=1;i<=n-1;i++) {int l,k;scanf("%d%d",&l,&k);edge[k][l]=true;}for(int i=1;i<=n;i++) ans=max(ans,dp(i));printf("%d",ans);return 0;
}