没有上司的舞会(树形dp)

Ural大学有N名职员，编号为1~N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 HiHi 给出，其中 1≤i≤N1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式

第一行一个整数N。

接下来N行，第 i 行表示 i 号职员的快乐指数HiHi。

接下来N-1行，每行输入一对整数L, K,表示K是L的直接上司。

输出格式

输出最大的快乐指数。

数据范围

1≤N≤6000,
−128≤Hi≤127

输入样例：

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

输出样例：

5

题意：第一行给出n，表示n位职工员，接下来n行，第i行给出第i为职员的快乐值，然后n-1行，每行两个数x，y，表示y时x的上司。求保证选出每位职员都没有直接上司的情况下快乐值的总和的最大值。

思路：n位职员的关系犹如一棵树，所以这是一个树形结构，采用树形dp:

用f[i][0]表示根节点为i的树，不选当前根结点i时，这棵树的最大快乐值。

用f[i][1]表示根节点为i的树，选当前根结点i时，这棵树的最大快乐值。

显然，不选当前根结点i时，i的子结点就可选可不选，每个子结点的最大快乐值就是这两种情况的max值，所以这棵树的最大快乐值就是所有子结点的最大值之和，即：

f[i][0]=max(f[j1][0],f[j1][1])+max(f[j2][0],f[j2][1])+max(f[j1][0],f[j1][1])+...     (jk表示i的第k个子结点)

选当前根结点i时，它的所有子结点就都不能选，所以以i为根的这棵树的最大快乐值，即：

f[i][1]=f[j1][0]+f[j2][0]+...       (jk表示i的第k个子结点)

因此，从树根开始，dfs往下dp即可，回溯时统计最大值。

注：找root时，（根据root的特征：因为root是整个树的根结点，所以root没有父结点），所以开始输入时，用vis数组标记一下每个结点是否有父结点，然后遍历所有结点，没有父结点即为root。

完整代码：

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn=6010;

int f[maxn][2],h[maxn];
int e[maxn],nex[maxn],head[maxn],cnt;
bool vis[maxn];

void addEdge(int u,int v)
{
    e[cnt]=v;
    nex[cnt]=head[u];
    head[u]=cnt++;
}

void dfs(int u)
{
    f[u][0]=0;
    f[u][1]=h[u];
    for(int i=head[u];~i;i=nex[i]){
        int v=e[i];
        dfs(v);
        f[u][0]+=max(f[v][0],f[v][1]);
        f[u][1]+=f[v][0];
    }
}

int main()
{
    int n;
    cin>>n;
    memset(head,-1,sizeof head);
    for(int i=1;i<=n;i++){
        cin>>h[i];
    }
    n--;
    while(n--){
        int x,y;
        cin>>x>>y;
        addEdge(y,x);
        vis[x]=true;        
    }
    int root=1;
    while(vis[root]) root++;
    dfs(root);
    cout<<max(f[root][0],f[root][1])<<endl;
    return 0;
}