没有上司的舞会

【题目描述】

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

【输入描述】

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

【输出描述】

输出最大的快乐指数。

【样例输入】

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

【样例输出】

5

源代码：

#include<cstdio>
#include<algorithm>
using namespace std;
int n,t1,t2,root=1,f[6001][2]={0},father[6001]={0}; //数组范围！不要吝惜空间！
void DP(int t) //树形DP。
{
    for (int a=1;a<=n;a++)
      if (father[a]==t) //判断是否为其直属部下。
      {
          DP(a); //递归到叶节点，后回溯求解。
          f[t][1]+=f[a][0]; //编号为t的职员若来，则应加上他的直属部下不来的解。
          f[t][0]+=max(f[a][1],f[a][0]); //若不来，则在他的直属部下来与不来的解中取最优值。
      }
}
int main()
{
    scanf("%d",&n);
    for (int a=1;a<=n;a++)
      scanf("%d",&f[a][1]); //编号为a的职员来时增加的欢乐度。
    scanf("%d%d",&t1,&t2);
    while (t1)
    {
        father[t1]=t2;
        scanf("%d%d",&t1,&t2);
    }
    while (father[root]) //寻找根节点。
      root=father[root];
    DP(root);
    printf("%d",max(f[root][1],f[root][0]));
    return 0;
}