HDU 1520 Anniversary party -- 樹形dp 好題目

/*
	http://acm.hdu.edu.cn/showproblem.php?pid=1520 Anniversary party
	比較典型的樹形dp，好題目!!!
	給一棵樹，要求取某一個節點（每一個點有權值）時不能取對應的父節點和子節點，問滿足條件的方案權值最大爲多少。
	定義狀態dp[i][0]表示不取i節點時的最大權值，dp[i][1]表示取i節點時的最大權值。
	所以
	dp[father][0] = max(dp[father][0] + dp[child][1] , dp[father][0] + dp[child][0]) 
	dp[father][1] = dp[father][1] + dp[child][0] ;
	在題目中，我們設一個0節點作爲整個“森林”（因爲不確定是不是樹）的root節點；
	我們只需要將結果遞推到dp[0][0]即可。
	題目中對數據的輸入沒有描述清楚，其實有多組測試數據，兩組之間有0 0隔開，最後還有一個0結束，否則將無限TLE
*/


#pragma comment(linker, "/stack:64000000")
#define _CRT_SECURE_NO_DEPRECATE

#include <queue>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
using namespace std;
#define CLR(c,v) memset(c,v,sizeof(c))

template <typename _T>
_T Max(_T a , _T b){
	return (a>b)?(a):(b);
}
template <typename _T>
_T Max(_T a , _T b, _T c){
	return (a>Max(b,c))?(a):(Max(b,c));
}
template <typename _T>
_T Min(_T a , _T b){
	return (a<b)?(a):(b);
}
template <typename _T>
_T Min(_T a , _T b, _T c){
	return (a<Min(b,c))?(a):(Min(b,c));
}

const int inf    = -(1<<30);
const int INF    =  (1<<30);
const double eps =  1e-8;
const int M      =  6e4 +10;
vector <int > p[M];


int f[M];
int dp[M][2];

void dfs(int cur){
	int size = p[cur].size();
	for(int i = 0 ; i < size ; i++){
		int child = p[cur][i];
		dfs(child);
		int father = cur;
		dp[father][0] += Max(dp[child][0],dp[child][1]);
		dp[father][1] += dp[child][0];
	}
}

int main(){
	//freopen("in.txt","r",stdin);
	int n,v;
	while(cin >> n && n!=0){
		CLR(f,0);
		CLR(dp,0);
		p[0].clear();
		for (int i = 1; i <= n ; i++){
			scanf("%d" , &v);
			f[i] = i;
			dp[i][1] = v;
			if(!p[i].empty())p[i].clear();
		}
		for (int a,b ,i=1; scanf("%d%d",&a,&b) && a!=0&&b!=0 ; i++){
			p[b].push_back(a);
			f[a] = b;
		}
		for (int i = 1 ; i <= n ; i++){ // 把森林或樹的root加入0節點下
			if(f[i] == i)
				p[0].push_back(i);
		}
		dfs(0);
		cout << dp[0][0] << endl;
	}
	return 0;
}