POJ 2342 Anniversiry Party（TYVJ1052 没有上司的舞会）

题意：

P1052 没有上司的舞会

描述

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

输入格式

第一行一个整数N。(1<=N<=6000)

接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)

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

最后一行输入0,0。

输出格式

输出最大的快乐指数。

测试样例1

输入

7

1

1

1

1

1

1

1

1 3

2 3

6 4

7 4

4 5

3 5

0 0

输出

5

思路：

树形DP

f[x][0]+=max(f[v[i]][1],f[v[i]][0]),f[x][1]+=f[v[i]][0];

//By: Sirius_Ren
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,xx,yy,f[6005][2],fa[6005],r,v[6005],tot=0,first[6005],nxt[6005];
bool vis[6005];
void add(int x,int y){v[tot]=y;nxt[tot]=first[x];first[x]=tot++;}
void dfs(int x){
    for(int i=first[x];~i;i=nxt[i])
        dfs(v[i]),f[x][0]+=max(f[v[i]][1],f[v[i]][0]),f[x][1]+=f[v[i]][0];
}
int main()
{
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=n;i++) scanf("%d",&f[i][1]);
    for(int i=1;i<n;i++) scanf("%d%d",&xx,&yy),add(yy,xx),vis[xx]=1;
    for(int i=1;i<=n;i++)if(!vis[i])r=i;
    dfs(r);
    printf("%d\n",max(f[r][0],f[r][1]));
}

一起 刷刷刷ing