Anniversary party（树形DP入门）

HDU-1520

#include<iostream>
#include<algorithm>
#include<vector>
#define ll long long
#define mem(a) memset(a,0,sizeof(a))
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;

const int N=6005;
int n,value[N],dp[N][2];//dp[i][1]第i个节点参加聚会能得到的最优值 
vector<int>a[N];		// dp[i][0]第i个节点不参加聚会能得到的最优值 
//用a[i]来存第i个节点的子节点 
void dfs(int fa){
	dp[fa][1]=value[fa];
	for(int i=0;i<a[fa].size();i++){
		int ch=a[fa][i];
		dfs(ch);
		dp[fa][1]+=dp[ch][0]; 
		dp[fa][0]+=max(dp[ch][0],dp[ch][1]);
	}
}
int main()
{
	int ch,fa,find_f[N];//find_f[]数组用来找根节点 
	while(cin>>n){
		for(int i=1;i<=n;i++){
			cin>>value[i];
			dp[i][1]=dp[i][0]=0;
			a[i].clear();
			find_f[i]=0;
		}
		while(cin>>ch>>fa){
			if(ch==0&&fa==0)break;
			a[fa].push_back(ch);
			find_f[ch]++;
		}
		for(int i=1;i<=n;i++){
			if(!find_f[i]){//当find_f[i]==0时，即找到根节点，进行搜索； 
				dfs(i);
				cout<<max(dp[i][1],dp[i][0])<<endl;
				break;
			}
		}	
	}

	return 0;
}