【CODEVS1380】没有上司的舞会

Description

      Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

Input

第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。

Output

输出最大的快乐指数。

Sample Input

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

Sample Output

5

HINT

各个测试点1s

 

#include<iostream>
using namespace std;
int l[6010]={0};//判断根节点 
int v[6010]={0};
int fc[6010];
int f[6010][2];
struct poitype{
    int next,to;
}e[6010]; 
int cur=0,n,ch,fa;//cur为总边数 
void insert(int u,int v)//插入一条边 
{
    ++cur;
    e[cur].to=u;
    e[cur].next=fc[v];
    fc[v]=cur;
}
void dp(int x)//0表示不参加,1表示参加 
{
    f[x][0]=0;f[x][1]=v[x];
    for (int i=fc[x];i!=0;i=e[i].next)
    {
        int ne=e[i].to;
        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;
}