[poj2342]Anniversary party树形dp入门

题意：选出不含直接上下司关系的最大价值。

解题关键：树形dp入门题,注意怎么找出根节点，运用了并查集的思想。

转移方程：dp[i][1]+=dp[j][0];/i是j的子树

　　　　　dp[i][0]+=max(dp[j][0],dp[j][1]);

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=7000;
vector<int>G[maxn];
int father[maxn];
int dp[6005][2];
void dfs(int u,int fa){
    int s=(int)G[u].size();
    for(int i=0;i<s;i++){
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][0],dp[v][1]);
    }
}
int main(){
    int n;
    while(cin>>n){
        memset(father, -1, sizeof father);
        int a,b;
        for(int i=1;i<=n;i++){
            cin>>dp[i][1];
        }
        int root=1;
        for(int i=0;i<n-1;i++){
            cin>>a>>b;
            G[b].push_back(a);
            father[a]=b;
            root=b;
        }
        cin>>a>>b;
        while(father[root]!=-1) root=father[root];
        dfs(root,0);
        int ans=max(dp[root][0],dp[root][1]);
        cout<<ans<<"
";
    }
    return 0;
}