标签:
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
第一行一个整数N。(1<=N<=6000)
接下来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都是以数组的形式顺序运算,但树形dp是以树的形式递归运算,给我的第一感觉像是dfs中用状态转移方程。。。
由所给数据建树,从根节点开始递归,每个节点储存 有参加 和 没参加 两种情况的最大快乐值。状态转移方程为:
pe[x].ha+=pe[i].k;//有参加的最大值为:本身的快乐值加上下属没参加的快乐值总和;
pe[x].k+=max(pe[i].ha,pe[i].k);//没参加的最大值为:下属没参加的快乐值总和和下属有参加的快乐值总和的较大者;
//.ha为有参加的快乐值总和(将初始快乐值录入) .k为没参加的快乐值总和(初始为0);
//代码如下
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
struct p{
int fa,so[1000],ha,k;
}pe[10000];
int n,v[6500];
int tree(int x){
v[x]=1;
for(int i=1;i<=n;i++)
if(v[i]==0&&pe[i].fa==x){//判断其为下属
tree(i);//递归
pe[x].ha+=pe[i].k;
pe[x].k+=max(pe[i].ha,pe[i].k);//状态转移方程
}
return 0;
}
int main(){
int i,j=1,l,k,tot;
cin>>n;
for(i=0;i<n;i++) scanf("%d",&pe[i+1].ha);
while(cin>>l&&cin>>k){
if(l==0&&k==0) break;
pe[l].fa=k;
}
i=1;
while(pe[i].fa!=0) i=pe[i].fa;//寻找根节点
tree(i);
printf("%d\n",max(pe[i].ha,pe[i].k));
return 0;
}
标签:
原文地址:http://www.cnblogs.com/wengsy150943/p/5769733.html