树形dp poj2342

#include <iostream>
#include <string.h>
#include <stdio.h>
//#pragma comment(linker, "/STACK:10240000,10240000")//扩栈
int n;
bool vis[6005];
int dp[6005][2],pre[6005];
using namespace std;
void tree_dp(int x)
{
    vis[x]=1;
    for(int i=1;i<=n;i++)
    {
        if(!vis[i]&&pre[i]==x)
        {
            tree_dp(i);
            dp[x][1]+=dp[i][0];//boss来了，下属没来
            dp[x][0]+=max(dp[i][0],dp[i][1]);//boss没来，下属可能来也可能没来
        }
    }
}
/*int find(int x) { int r=x; while(r!=pre[r]) r=pre[r]; int i=x,j; while(pre[i]!=r) { j=pre[i]; pre[i]=r; i=j; } return r; }*/
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int a,b,q=0,t=0;
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        memset(pre,0,sizeof(pre));
        for(int i=1;i<=n;i++)
            {
                scanf("%d",&dp[i][1]);
                //pre[i]=i;//poj的数据这里有bug，要不然我早就过了，可能会出现不是1到n的数，比如4 1,1,1,1 2，7 5， 2就没法解决，但是我个人认为这个地方应该没问题的，是数据的问题 
            }
           while(scanf("%d%d",&a,&b)!=EOF)
            {
                if(a==0||b==0)
                    break;
                pre[a]=b;
                 q=a;
            }
            while(q!=pre[q])
            {
                q=pre[q];
            }
            tree_dp(q);
            printf("%d\n",max(dp[q][0],dp[q][1]));
    }
    return 0;
}