描述
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
输入格式
第一行一个整数N。(1<=N<=3000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。
输出格式
输出最大的快乐指数。
样例输入
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
样例输出
5
题解:树形DP入门题(很久以前的大坑还没有补。。TAT),树形DP首先我们需要建树,我们在读入时,对于每一个给定的关系,连接一条以直接上司为起点,下属为终点的边,并计算每一个点的入度(入度:有N条边的终点是该点,那么这个点的入度就是N),显然根节点就是入度为0的那个点。找到根节点,接着开始讨论方程,我们可以试想一下,f[i][0]表示编号为i的人不来参加的话,以他为根节点的子树能获得的最大快乐值是多少,f[i][1]则表示如果编号为i的人来参加的话,以i为根节点的子树获得的最大快乐值是多少。 由于直接上司和下属不能同时参加,那么我们就可以很快得出递推方程。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#define LiangJiaJun main
using namespace std;
int f[3004][2],n,a[3004],l[3004],h[3004],ne=1;
struct edge{
int to,next;
}e[8004];
void insert(int u,int v){
e[++ne].to = v;
e[ne].next = h[u];
h[u] = ne;
}
void dp(int x){
f[x][1]=a[x];f[x][0]=0;
for(int i=h[x];i;i=e[i].next){
dp(e[i].to);
f[x][0]=max(f[x][0],max(f[x][0]+f[e[i].to][1],f[x][0]+f[e[i].to][0]));
f[x][1]=max(f[x][1],f[x][1]+f[e[i].to][0]);
}
}
int LiangJiaJun(){
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++){
int u,v;
scanf("%d%d",&u,&v);
++l[u];insert(v,u);
}
for(int i=1;i<=n;i++)if(!l[i]){
dp(i);
printf("%d\n",max(f[i][0],f[i][1]));
return 0;
}
return 0;
}