0715-树形dp-POJ2342

POJ试题：http://poj.org/problem?id=2342

题目描述

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

输入格式

第一行一个整数 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

输出

5

这是一道很典型的树的最大独立集的问题

从n个点中选取一些点，使其两两之间不相连接，且点权最大（这里就是员工们的快乐值了）

无关乎谁是boss谁是employee，直接搞

对于每个点而言，都有两种状态，选或者不选，如果选了，则它儿子必须不选

如果不选，则它儿子可选可不选

#include <map>
#include <cstring>
#include <iostream>
#include<vector>
#include<cstdio>
#include<cmath>
#include<algorithm> 
#define maxn 6009
using namespace std;
vector<int> q[maxn];
int a[maxn],n;
int d[maxn][3];

void dfs(int u,int fa){
	if(d[u][0]&&d[u][1]) return;
	d[u][1]=a[u];
	int i,j,k;
	for(i=0;i<q[u].size();++i){
		int v=q[u][i];
		if(v!=fa){
			dfs(v,u);
			d[u][0]+=max(d[v][0],d[v][1]);
			d[u][1]+=d[v][0];
		}
	}
}
int main()
{
	scanf("%d",&n);
	int i,j,k;
	for(i=1;i<=n;++i){
		scanf("%d",&a[i]);
	}
	int l;
	scanf("%d%d",&l,&k);
	while(l!=0&&k!=0){
		q[l].push_back(k);
		q[k].push_back(l);
		scanf("%d%d",&l,&k);
	}
	dfs(1,-1);
	printf("%d",max(d[1][0],d[1][1]));
	return 0;
}