树的最大独立子集

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
typedef long long ll;
const int maxn=6100;
vector<int>G[maxn];
int dp[maxn][2];        //0表示不取这个数，1表示取这个数
int f[maxn];            //寻找根
int a[maxn];

void dfs(int s){
    int len=G[s].size();
    dp[s][1]=a[s];
    for(int i=0;i<len;i++)
        dfs(G[s][i]);
    for(int i=0;i<len;i++){
        dp[s][0]+=max(dp[G[s][i]][0],dp[G[s][i]][1]);
        dp[s][1]+=dp[G[s][i]][0];
    }
}

int main(){
    int n;
    int x,y;
    while(scanf("%d",&n)!=EOF){
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            f[i]=0;
            G[i].clear();                   //不要忘记初始化
        }
        mem0(dp);
        while(scanf("%d%d",&x,&y)==2){
            if(x==0&&y==0)
                break;
            f[x]=1;
            G[y].push_back(x);          //不用双想建边,只是上司来了，下属不能来
        }                               //建边的时候如果是双向关系，则双向建边
        int a;
        for(int i=1;i<=n;i++){
            if(f[i]==0){
                a=i;
                break;
            }
        }
        dfs(a);
        printf("%d\n",max(dp[a][0],dp[a][1]));
    }
    return 0;
}