【CODEVS1380】没有上司的舞会

#include<iostream>
using namespace std;
int l[6010]={0};//判断根节点 
int v[6010]={0};
int fc[6010];
int f[6010][2];
struct poitype{
    int next,ch;
}poi[6010]; 
int cur=0,n,ch,fa;//cur为总边数 
void insert(int ch,int fa)//插入一条边 
{
    ++cur;
    poi[cur].ch=ch;//编号的儿子 
    poi[cur].next=fc[fa]; //若fc[fa]有值,则fa出现过两次,即ch有兄弟,所以next里存的是ch的兄弟 
    fc[fa]=cur;
}
void dp(int x)//0表示不参加,1表示参加 
{
    f[x][0]=0;f[x][1]=v[x];
    for (int i=fc[x];i!=0;i=poi[i].next)
    {
        int ne=poi[i].ch;
        dp(ne);
        f[x][1]+=f[ne][0]; //下属都不参加 
        f[x][0]=f[x][0]+max(f[ne][1],f[ne][0]);
    }
}
int main()
{
    cin>>n;
    for (int i=1;i<=n;i++) cin>>v[i];
    while (cin>>ch>>fa && ch)
    {
        l[ch]++;
        insert(ch,fa);
    }
    for (int i=1;i<=n;i++)
    {
        if (l[i]==0)
        {
            dp(i);
            cout<<max(f[i][0],f[i][1]);
            break;
        } 
    }
    return 0;
}