[tyvj1052]没有上司的舞会

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

第一行一个整数N。(1<=N<=3000)
接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行,每行输入一对整数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

题解:树形DP入门题(很久以前的大坑还没有补。。TAT),树形DP首先我们需要建树,我们在读入时,对于每一个给定的关系,连接一条以直接上司为起点,下属为终点的边,并计算每一个点的入度(入度:有N条边的终点是该点,那么这个点的入度就是N),显然根节点就是入度为0的那个点。找到根节点,接着开始讨论方程,我们可以试想一下,f[i][0]表示编号为i的人不来参加的话,以他为根节点的子树能获得的最大快乐值是多少,f[i][1]则表示如果编号为i的人来参加的话,以i为根节点的子树获得的最大快乐值是多少。 由于直接上司和下属不能同时参加,那么我们就可以很快得出递推方程。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#define LiangJiaJun main
using namespace std;
int f[3004][2],n,a[3004],l[3004],h[3004],ne=1;
struct edge{
    int to,next;
}e[8004];
void insert(int u,int v){
     e[++ne].to = v;
     e[ne].next = h[u];
     h[u] = ne;
}
void dp(int x){
     f[x][1]=a[x];f[x][0]=0;
     for(int i=h[x];i;i=e[i].next){
         dp(e[i].to);
         f[x][0]=max(f[x][0],max(f[x][0]+f[e[i].to][1],f[x][0]+f[e[i].to][0]));
         f[x][1]=max(f[x][1],f[x][1]+f[e[i].to][0]);
     }
}
int LiangJiaJun(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        ++l[u];insert(v,u);
    }
    for(int i=1;i<=n;i++)if(!l[i]){
        dp(i);
        printf("%d\n",max(f[i][0],f[i][1]));
        return 0;
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值