没有上司的舞会

problem 1352 没有上司的舞会

---

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题，所以我才第一个做了它。
亮点就在于变长数组vector，他可以不用担心树的长度，很好用；
在这里尤其适用；

---

代码

水平有限，只实现了记忆化搜索，转动归懒得写了；

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstdlib>
using namespace std;

int n, A,B,dp[100005][3],i,a[100005];
bool du[100005];
vector <int> ch[100005];//这就是变长数组了

int dfs(int pos,int ok)
{
   if(ch[pos].size()==0)//如果当前是叶节点，那么就执行回溯条件
    {
        if(ok) return 0;
        return max(0,a[pos]);
    }

   if(ok)//如果ok，也就是其父是否被选为真；那么其子就可选；此点不可选；
    {
        for(int i=0;i<ch[pos].size();i++)
         dp[pos][ok]+=(dp[ch[pos][i]][0]?dp[ch[pos][i]][0]:dfs(ch[pos][i],0));
    }

   if(!ok)//如果!ok，也就是其父是否被选为假；那么当前节点可以选也可不选
    {   int p=0;
        for(int i=0;i<ch[pos].size();i++)
         p+=(dp[ch[pos][i]][0]?dp[ch[pos][i]][0]:dfs(ch[pos][i],0));//不选该点

        int q=a[pos];
        for(int i=0;i<ch[pos].size();i++)//选择该点
         q+=(dp[ch[pos][i]][1]?dp[ch[pos][i]][1]:dfs(ch[pos][i],1));
        return max(p,q);
    }
    return dp[pos][ok];

}

int main()
{
    scanf("%d",&n);
    for(i=1;i<=n;i++)
     scanf("%d",&a[i]);
    for(i=1;i<n;i++)
      {
        scanf("%d%d",&A,&B);
        ch[B].push_back(A);
        du[A]=1;
      }
    for(i=1;i<=n;i++)
       if(du[i]==0) break;
    cout<<dfs(i,0);

     return 0;
}