poj 2342 Anniversary party_经典树状dp

最新推荐文章于 2020-08-14 09:29:12 发布

原创最新推荐文章于 2020-08-14 09:29:12 发布 · 800 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#acm-icpc #dp #动态规划

动态规划专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一个关于Ural大学职员参加会论的问题。通过建立一棵表示职员间隶属关系的树，并为每个职员分配一个快乐指数，目标是确定哪些职员应该参加会议以最大化总体快乐指数，同时确保没有职员与其直接上司同时参会。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：Ural大学有n个职员，1~N编号，他们有从属关系，就是说他们关系就像一棵树，父节点就是子节点的直接上司，每个职员有一个快乐指数，现在要开会，职员和职员的直接上司不能同时开会，问怎才能使开会的快乐指数最高。

思路：用树状dp，dp[i][0]为不参加会议，dp[i][1]为参加会议

#include<iostream>
#include<cstdio>
#include<vector>
#define N 6010
using namespace std;
vector<int> v[N];
int dp[N][2],hp[N],mark[N];
int max(int a,int b){return a>b?a:b;}
void dfs(int k)
{
	int i,len=v[k].size();
	dp[k][0]=0;
	dp[k][1]=hp[k];
	if(len==0)
		return ;
	for(i=0;i<len;i++)
	{
		dfs(v[k][i]);
		dp[k][0]+=max(dp[v[k][i]][1],dp[v[k][i]][0]);//不参加会议，下属有可以参加会议，也可以不参加会议
		dp[k][1]+=dp[v[k][i]][0];//参加会议，下属不能参加会议

	}
}
int main()
{
	int n,a,b,i,k;
	while(scanf("%d",&n)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%d",&k);
			hp[i]=k;
			v[i].clear();
		}
		memset(mark,0,sizeof(mark));
		while(scanf("%d%d",&b,&a)&&(a||b))
		{
			v[a].push_back(b);
			mark[b]=1;
		}
		for(i=1;i<=n;i++)//找出根节点
			if(!mark[i])
				break;
		dfs(i);
		printf("%d\n",max(dp[i][0],dp[i][1]));
	}
	return 0;
}