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

题目描述

某大学有 nn个职员,编号为 1n。

他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。

现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数 ri,但是呢,如果某个职员的直接上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。

所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。

输入格式

输入的第一行是一个整数 n。

第 2 到第 (n+1) 行,每行一个整数,第 (i+1) 行的整数表示 i 号职员的快乐指数ri

第 (n + 2) 到第(2n+1) 行,每行输入一对整数 l, k代表 k 是 l 的直接上司。

输出格式

输出一行一个整数代表最大的快乐指数。

输入输出样例

输入 #1
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出 #1
5

说明/提示

数据规模与约定

对于100% 的数据,保证  1n6×10^3,128ri127,1l,kn,且给出的关系一定是一棵树。

很显然,这是一个树形dp(我也没学过,老师说的QWQ),那么首先要找状态转移方程,每个节点有两个需要dp的条件,一个是选或者不选,另一个是当前位置,所以开一个二维数组来记录数据。第一维表示状态,第二维记录位置。所以思路就很清晰了,详情见代码

代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
int ru[12010],po[12010],ne[12010],f[5][12010];//开数组,ru数组的含义是有多少个父亲节点,po是父节点编号,next就是子节点,二维的f数组则记录最值答案
int n,a,b,root;//root是根结点
void dp(int x)
{
for(int i=po[x];i;i=ne[i])//这里不懂,存疑!!!!!!!!!!!!!!!!!!!!!!
{
dp(i);
f[1][x]=max(max(f[1][x],f[1][x]+f[0][i]),f[0][i]);//其实这里就是分类讨论,讨论有几种情况,这里分别是:1.只选当前的节点。2.选当前节点并且加上不选的子节点的最值。3.赋值不选的那个子节点位置上的最值。可能会有疑问,我这个大的分类就是选这个节点啊,为什么还能不选这个节点,只选子节点。我也有这个疑问。。。。。。。。。。。。。。。。。。
f[0][x]=max(max(f[0][x],f[0][x]+f[1][i]),max(f[1][i],f[0][i]));//这里多了一种情况:1.自己不选。2.自己不选选子节点。3.只选子节点。4.不选子节点。我也晕了
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&f[1][i]);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&b,&a);
ru[b]++;
ne[b]=po[a];//这里不懂
po[a]=b;
}
for(int i=1;i<=n;i++)
{
if(ru[i]==0)//如果一个节点没有父亲节点,那肯定是根结点
{
root=i;//记录根结点的位置
break;
}
}
dp(root);//从根结点开始dp,因为函数里是先递归,再写内容,所以会从最下面的叶子节点开始,那么到最后数据都在根节点里
printf("%d",max(f[1][root],f[0][root]));//同上
return 0;
}

猜你喜欢

转载自www.cnblogs.com/57xmz/p/12825010.html