题目描述
某大学有N个职员,编号为1~N。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。
输入输出格式
输入格式:
第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0
输出格式:
输出最大的快乐指数。
输入输出样例
输入样例#1:
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
输出样例#1:
5
这道题可以说是树形DP的入门题。设二维数组f[i][0]表示不选i
参加舞会时i下属的最大快乐指数,f[i][1]表示选i参加舞会时下
属的最大快乐指数。
可得出DP转移方程
f[x][0]+=max(f[a[x][i]][0],f[a[x][i]][1]);
f[x][1]+=f[a[x][i]][0];
由于DP是一个递归的过程,所以用void来写,完整的void如下
void dp(int x){
for(int i=0;i<a[x].size();i++){
int use=a[x][i];
dp(use);
f[x][0]+=max(f[use][0],f[use][1]);
f[x][1]+=f[use][0];
}
}
本题是从树根开始的树形DP,所以还需找到树根。找到树根,
直接做一遍DP,输出max(f[root][1],f[root][0])即可。
code:
#include<bits/stdc++.h>
#include<vector>
using namespace std;
int i,j,k,n,m,tot,ans,root,f[6005][2],fa[6005];
vector<int> a[6005];
int read(){
char c;while(c=getchar(),(c<'0'||c>'9')&&c!='-');
int x=0,y=1;if(c=='-') y=-1;else x=c-'0';
while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';
return x*y;
}
int maxx(int a,int b){return a>b?a:b;}
void dp(int x){
for(int i=0;i<a[x].size();i++){
int use=a[x][i];
dp(use);
f[x][0]+=maxx(f[use][0],f[use][1]);
f[x][1]+=f[use][0];
}
}
int main()
{
n=read();
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=n;i++) f[i][1]=read();
while(1>0){
int x=read(),y=read();
if(x==0&&y==0) break;
fa[x]=y; a[y].push_back(x);
}
for(int i=1;i<=n;i++) if(fa[i]==i){
root=i;break;
}
dp(root);
printf("%d",maxx(f[root][1],f[root][0]));
return 0;
}