没有上司的舞会(树形dp)

Ural大学有N名职员,编号为1~N。

他们的关系就像一棵以校长为根的树,父节点就是子节点的直接上司。

每个职员有一个快乐指数,用整数 Hi 给出,其中 1≤i≤N。

现在要召开一场周年庆宴会,不过,没有职员愿意和直接上司一起参会。

在满足这个条件的前提下,主办方希望邀请一部分职员参会,使得所有参会职员的快乐指数总和最大,求这个最大值。

输入格式
第一行一个整数N。

接下来N行,第 i 行表示 i 号职员的快乐指数Hi。

接下来N-1行,每行输入一对整数L, K,表示K是L的直接上司。

最后一行输入0,0。

输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,
−128≤Hi≤127
输入样例:
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
输出样例:
5

解题思路:

任何一个人的来或不来可以看作一种决策,那么状态就是在某个人来的时候或者不来的时候,以他为根的子树能有的最大活跃总值。分别可以用dp[i][1]和dp[i][0]表示第i个人来和不来。

所以可以得到状态转移方程:

第i个人来:dp[i][1] += dp[child][0](他的直接下属一定不能来)

第i个人不来:dp[i][0] += max ( dp[child][0], dp[child][1])(他的直接下属可来可不来,取较大值)

#include
using namespace std;

const int N=6010;
vector G[N];
int happy[N],dp[N][2];
int n;
bool st[N];

void dfs(int u){
	dp[u][0]=0;
	dp[u][1]=happy[u];
	
	for(int i=0;i>n;
	for(int i=1;i<=n;i++) cin>>happy[i];
	
	for(int i=1;i>a>>b;
		G[b].push_back(a);
		st[a]=true;
	}
	
	int root=1;
	while(st[root]){
		root++;
	}
	
	dfs(root);
	cout<

你可能感兴趣的:(树形dp)