码迷,mamicode.com
首页 > 其他好文 > 详细

树形dp-没有上司的舞会

时间:2016-08-14 11:34:14      阅读:121      评论:0      收藏:0      [点我收藏+]

标签:

题目描述 Description

      Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。

输入描述 Input Description

第一行一个整数N。(1<=N<=6000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。

输出描述 Output Description

输出最大的快乐指数。

样例输入 Sample Input

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

样例输出 Sample Output

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;
}

树形dp-没有上司的舞会

标签:

原文地址:http://www.cnblogs.com/wengsy150943/p/5769733.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!