LOJ10160周年纪念晚会

题目描述

Ural 州立大学的校长正在筹备学校的 80 周年纪念聚会。由于学校的职员有不同的职务级别，可以构成一棵以校长为根的人事关系树。每个资源都有一个唯一的整数编号，从 1 到 N 编号，且对应一个参加聚会所获得的欢乐度。为使每个职员都感到快乐，校长设法使每个职员和其直接上司不会同时参加聚会。

你的任务是设计一份参加聚会者的名单，使总欢乐度最高。

输入格式

第一行是一个整数 N；

接下来 N 行对应 N 个职员的欢乐度，第 ii 行的一个整数为第 i 个职员的欢乐度 pi​；

接着是学校的人事关系树，每一行格式为 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

数据范围与提示

对于100% 的数据，1≤N≤6000,−128≤pi​≤127。

——————————————————————————————————————————————————

没有上司的舞会，简单的树形动归。

f[u][0]：表示当前点不参加舞会时，当前子树上的最大欢乐值

f[u][1]：表示当前点参加舞会时，当前子树上的最大欢乐值

f[u][0]=sum( max( f[v][0] , f[v][1] ) )

f[u][1]=sum( f[v][0] ) + w[u]

——————————————————————————————————————————————————

#include<bits/stdc++.h>
using namespace std;
const int maxn=6010;
int n;
struct edge
{
    int u,v,nxt;
}e[maxn];
int head[maxn],js;
void addage(int u,int v)
{
    e[++js].u=u;e[js].v=v;
    e[js].nxt=head[u];head[u]=js;
}
int w[maxn],root;
bool rd[maxn];
int f[maxn][2];
void dp(int u,int fa)
{
    f[u][0]=0;f[u][1]=w[u];
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(v!=fa)
        {
            dp(v,u);
            f[u][0]+=max(f[v][0],f[v][1]);
            f[u][1]+=f[v][0];
        }
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i)scanf("%d",&w[i]);
    for(int u,v,i=1;i<n;++i)
    {
        scanf("%d%d",&v,&u);
        addage(u,v);
        rd[v]=1;
    }
    for(int i=1;i<=n;++i)
        if(rd[i]==0)
        {
            root=i;
            break;
        }
    dp(root,0);
    cout<<max(f[root][0],f[root][1]);
    return 0;
}