codevs 没有上司的舞会

1380 没有上司的舞会
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
题解
题目描述 Description
Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

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

输出描述 Output Description
输出最大的快乐指数。

样例输入 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

应该算的上是最简单的树状dp了，思路很简单，但是还是敲了二十分钟，为自己的手速感到汗颜。

#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;
vector<int>tree[6005];
int num[6005][2],top,n;//定义状态num[i][j]，j=0时表示i节点的父节点没有被选中，j=1时表示i节点的父节点被选中了
int dp(int k,int s)
{
    if ( num[s][k] ) return num[s][k];//如果不为零表示已经计算过了
    int sum1 = tree[s][0],sum2 = 0;//sum1表示选择当前节点的结果，sum2表示不选择当前节点的结果
    if ( !k )
    {
        for(int i = 1; i < tree[s].size(); i ++ )//子节点都不可选择
        {
            sum1 += dp(1,tree[s][i]);
        }
    }
    for(int i = 1; i < tree[s].size(); i ++ )//子节点可选可不选
    {
        sum2 += dp(0,tree[s][i]);
    }
    if ( !k )
    {
        num[s][k] = sum2 > num[s][k] ? sum2 : num[s][k];
        num[s][k] = sum1 > num[s][k] ? sum1 : num[s][k];
    }
    else num[s][k] = sum2 > num[s][k] ? sum2 : num[s][k];
    return num[s][k];
}
int main()
{
    int s,a,b,k = 0;
    scanf("%d",&n);
    for(int i = 1; i <= n; i ++ )
    {
        scanf("%d",&s);
        tree[i].push_back(s);
        top = i;
    }
    while ( scanf("%d%d",&a,&b) != EOF )
    {
        if ( !a && !b ) break;
        if ( !k ) top = b;
         else if ( a == top ) top = b;
         tree[b].push_back(a);
         k++;
    }
    memset(num,0,sizeof(num));
    printf("%d\n",dp(0,top));
    return 0;
}