POJ2342 Anniversary party 【树形dp】

题目链接：https://vjudge.net/problem/POJ-2342

 

题意：

一些人去参加party，每个人都有一个愉悦值，但是如果他的直接上司也在party就会变的很沮丧。为了避免这种情况，现在给出n个人的愉悦值以及这些人之间的关系，问能达到的最高的愉悦值是多少？？

 

题解：

一道树形dp题，不妨设dp[i][0] dp[i][1] 分别为第i个人来或不来所得到的最高的愉悦值，则有：

 

dp[fa][1] += dp[i][0];
dp[fa][0] += max(dp[i][1], dp[i][0]); 
其中 fa 是 i 的直接上司

 

然后这道题就水过去了。。

 

代码：

 

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int size = 6005;
int n;
int dp[size][2], father[size], vis[size];

void tdp(int vert) {
	vis[vert] = 1;
	for ( int i = 1; i <= n; i ++ ) {
		if(!vis[i] && father[i] == vert ) {
			tdp(i);
			dp[vert][1] += dp[i][0];
			dp[vert][0] += max(dp[i][1], dp[i][0]);
		}
	}
}

int main() {
	while( scanf("%d", &n) != EOF ) {
		memset(dp, 0, sizeof(dp));
		memset(father, 0, sizeof(father));
		memset(vis, 0, sizeof(vis));

		for ( int i = 1; i <= n; i ++ ) scanf("%d", &dp[i][1]);

		bool fg = 1;
		int root;
		int c, f;

		while( scanf("%d %d", &c, &f), (c+f) ) {
			father[c] = f;
			if(root == c || fg) {
				root = f;
			}
		}

		while(father[root]) root = father[root];
		tdp(root);

		int mx = max(dp[root][0], dp[root][1]);
		printf("%d\n", mx);
	} 

	return 0;
}