树形DP 洛谷1352 没有上司的舞会

题目:

题目描述

某大学有N个职员,编号为1~N。他们之间有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。现在有个周年庆宴会,宴会每邀请来一个职员都会增加一定的快乐指数Ri,但是呢,如果某个职员的上司来参加舞会了,那么这个职员就无论如何也不肯来参加舞会了。所以,请你编程计算,邀请哪些职员可以使快乐指数最大,求最大的快乐指数。

输入输出格式

输入格式:

第一行一个整数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
输出样例#1: 
5















这道题是最基础的树形DP,运用了新新工具vector(以前从来没用过)

首先你要找到最上司(没有上司的那个人是大boss哟),然后分类讨论(数学学多了)

f[i][0]表示这个人不去舞会 反之则去

代码如下:

//By ChristalL 
#include
#include
#include
#include
#define MAXN 6010
using namespace std;
vector down[MAXN];
int n,Ri[MAXN],f[MAXN][2],boss;
int p[MAXN];
void dfs(int x)
{
	f[x][0]=0,f[x][1]=Ri[x];
	for(int i=0;i
这样就OK啦!



你可能感兴趣的:(树形DP 洛谷1352 没有上司的舞会)