树形DP入门2019-08-12

【题意】
公司有n个人，每个人有价值vi，有一天举办年会，每个人都可以参加，但有严格的等级制度，参加活动时，不能同时出现a和a的上司，问如何才能使总和最大。
【分析】
每个人只有去和不去两种状态，设f[i][0]和f[i][1]分别表示第i个人不参加和参加年会，获得的总的最大价值。
则状态转移方程为：
f[i][1] += f[j][0],
f[i][0] += max{f[j][0],f[j][1]};
其中j为i的孩子节点。
这样，从根节点root进行dfs，最后结果为max{f[root][0],f[root][1]}。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
const int maxn=10000+5;
using namespace std;
struct node{
	int fa;
	vector<int>ch;
}a[maxn];//结构体
int f[maxn][2],p[maxn];
void dfs(int x){//深搜树形DP
	p[x]=1;
	int s=a[x].ch.size();
	for(int i=0;i<s;i++){
		int k=a[x].ch[i];
		if(!p[k]){
			dfs(k);
			f[x][1]+=f[k][0];
			f[x][0]+=max(f[k][0],f[k][1]);
		}
	}
}
int main(){
	int k,m,n,x,y,root=1;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&f[i][1]);	
	while(scanf("%d%d",&x,&y)){
		if(x==0&&y==0)
			break;
		a[x].fa=y;	//存图
		a[y].ch.push_back(x);//存入结构体
	}
	for(int i=1;i<=n;i++)
		if(a[i].fa==0){//寻找根节点（CEO）
			root=i;
			break;
		}
	dfs(root);
	printf("%d\n",max(f[root][0],f[root][1]));//输出CEO去或不去的最大值
	return 0;
}