CodeVS.1380 没有上司的舞会 (树形DP)

CodeVS.1380 没有上司的舞会 (树形DP)

标签（空格分隔）： 动态规划

题意分析

题目描述

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

输入描述

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

题解

简单的树形dp，一句话解释就是：
如果rt参加，那rt的下属不能参加，如果rt不参加，rt的下属可以选择参加或者不参加，取其最大。

代码总览

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define nmax 10000
using namespace std;
struct edge{
    int to,next;
}edg[nmax];
int head[nmax],dp[nmax][2];
int happy[nmax];
int tot = 1,n;
bool isroot[nmax];
void add(int u,int v){
    edg[tot].to = v;
    edg[tot].next = head[u] ;
    head[u] = tot++;
}
void dfs(int rt){//0 represent not choose & 1 represent choose
    dp[rt][0] = 0;
    dp[rt][1] = happy[rt];
    for(int i = head[rt]; i!=-1; i = edg[i].next){
        int nxt = edg[i].to;
        dfs(nxt);
        dp[rt][0] += max(dp[nxt][0],dp[nxt][1]);
        dp[rt][1] += dp[nxt][0];

    }
}
int main()
{
    memset(head,-1,sizeof head);
    memset(isroot,true,sizeof isroot);
    scanf("%d",&n);
    for(int i = 1;i<=n;++i) scanf("%d",&happy[i]);
    while(true){
        int u,v;
        scanf("%d %d",&u,&v);
        if(u == 0 && v == 0) break;
        add(v,u);
        isroot[u] = 0;
    }
    int rt = 0;
    for(int i = 1;i<=n;++i) if(isroot[i] == true){rt=i;break;}
    dfs(rt);
    printf("%d\n",max(dp[rt][0],dp[rt][1]));
    return 0;
}