SDU程序设计思维与实践 Week13 没有上司的舞会【树形DP】

最新推荐文章于 2022-07-12 00:00:00 发布

Catmull

最新推荐文章于 2022-07-12 00:00:00 发布

阅读量110

点赞数

分类专栏： SDU程序设计思维与实践

本文链接：https://blog.csdn.net/weixin_43975891/article/details/106139176

版权

SDU程序设计思维与实践专栏收录该内容

46 篇文章 1 订阅

订阅专栏

没有上司的舞会

公司共有n（n≤6000）位员工。公司要举行一个舞会。为了让到会的每个人不受他的直接上司约束而能玩得开心，公司领导决定：如果邀请了某个人，那么一定不会再邀请他的直接的上司，但该人的上司的上司，上司的上司的上司……都可以邀请。已知每个人最多有唯一的一个上司。公司的每个人参加晚会都能为晚会增添一些气氛，求一个邀请方案，使气氛值的和最大。

输入描述

结点按 1～N 编号。
第一行为 N (1 ≤ N ≤ 6000) ，代表结点个数。
接下来 N 行分别代表每个人的气氛值 w[i]（-128 ≤ w[i] ≤ 127)。
接下来 N-1 行，每行两个数 L K，代表 K 是 L 的一个父节点。
输入有多组，以 0 0 结束。

输出描述

每组数据输出一个整数，代表气氛值总和的最大值。

思路分析

定义 $f [i] [0 / 1]$ 表示以 i 为节点的子树所能得到的最大的气氛值
f[i][0] 表示以 i 为子树且未选择 i 号节点；
f[i][1] 表示以 i 为子树且选择了 i 号节点
最终答案就是 $m a x (f [r o o t] [0], f [r o o t] [1])$
$\sum\{f[x][1],f[x][0]\}$
$f[i][1]=a_i+\sum f[x][0]$

代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxN=6010;
int head[maxN],tot,N,w[maxN],k,l,f[maxN][2],root;
bool isroot[maxN];
struct Edge{
	int u,v,next;
};
Edge e[maxN];
void init(int n){
	tot=0;
	for(int i=1;i<=n;i++){
		head[i]=0;
		isroot[i]=true;	
	} 
	memset(f,0,sizeof f);	
}
void addEdge(int u,int v){
	tot++;
	e[tot].u=u;
	e[tot].v=v;
	e[tot].next=head[u];
	head[u]=tot;
}
void dfs(int x){
	f[x][1]=w[x];
	for(int i=head[x];i;i=e[i].next){
		int y=e[i].v;
		dfs(y);
		f[x][0]+=max(f[y][1],f[y][0]);
		f[x][1]+=f[y][0];
	}
}
int main(){
	while(true){
		cin>>N;
		if(N==0) break;
		init(N);
		for(int i=1;i<=N;i++){
			cin>>w[i];
		}
		for(int i=1;i<=N-1;i++){
			cin>>l>>k;
			isroot[l]=false;
			addEdge(k,l); 
		}
		for(int i=1;i<=N;i++){
			if(isroot[i]) root=i; 
		}		
		dfs(root);
		cout<<max(f[root][0],f[root][1])<<endl;
	}
}