hdu 1520 Anniversary party 树状dp

最新推荐文章于 2024-05-22 17:32:19 发布

原创最新推荐文章于 2024-05-22 17:32:19 发布 · 230 阅读

0 ·

CC 4.0 BY-SA版权

——动态规划—— 同时被 2 个专栏收录

43 篇文章

订阅专栏

dfs

25 篇文章

订阅专栏

本文介绍了一种在大学庆典中选择参加派对人员的算法，该算法基于员工间的上下级关系形成的树状结构，通过动态规划求解在满足一定条件下的最大趣味值。

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

题意：在一个大学庆典中，从n个职工中选择一些人来参加派对，这n个人上司与下属关系形成一颗树，每个人都有一个对应的funny值，为了使派对更加有趣，要求上司与其直接下属不能同时参加派对，并使派对的funny值最大.

还以为是图，问题想复杂了

#include<cstdio>
#include<cstring>
#include<vector>
#define Max(a,b) a>b?a:b
using namespace std;
const int maxn = 6010;
vector<int> g[maxn];
int fa[maxn],vis[maxn];
int dp[maxn][2];
int n;
void tree_dp(int u)
{
	vis[u]=1;
	for(int i=0;i<g[u].size();i++)
	{
		int v = g[u][i];
		if(!vis[v])
		{
			tree_dp(v);
			dp[u][0]+=Max(dp[v][0],dp[v][1]);
			dp[u][1]+=dp[v][0];
		}
	}
}
int main()
{
	int a,b;
	while(scanf("%d",&n)!=EOF)
	{
			
	memset(dp,0,sizeof(dp));
	memset(fa,0,sizeof(fa));
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&dp[i][1]);
		g[i].clear();
	}
	
	int root,ans;
	while(scanf("%d%d",&a,&b)&&(a!=0||b!=0))
	{
		fa[a]=b;
		g[b].push_back(a);
		root=b;
	}
	while(fa[root])root=fa[root];
	
	tree_dp(root);
	ans=Max(dp[root][0],dp[root][1]);
	printf("%d\n",ans);
	
	}

}