poj 4045 (树形DP)

博客介绍了如何在POJ 4045问题中利用树形动态规划(Tree DP)求解。首先选定根节点,计算从根节点到所有子节点的距离和dp[i],并记录每个节点的子节点数量Node[i]。接着,计算除子节点外的其他节点到当前点的距离贡献,公式为dp1[v] += (dp1[u] + (dp[u] - dp[v] - Node[v] - 1) + n - Node[v] - 1)。这个过程涉及到了节点间的距离计算和子节点的排除处理。

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

先选一点为根节点找出所有父节点i到下面所有点距离和dp[i],该父节点下面有多少个点Node[i]。

然后求出所有节点的所有非子节点到该点的距离dp1[v]+=(dp1[u]+(dp[u]-dp[v]-Node[v]-1)+n-Node[v]-1)

dp[u]-dp[v]-Node[v]-1:u的子节点中除了v这一部分子节点到u的距离

n-Node[v]-1:非v的字节点的个数







#include<stdio.h>
#include<string.h>
#define N 50002
#define inf 0x3fffffff
int head[N],num,vis[N],dp[N],Node[N],dp1[N],n,I,R;
struct edge
{
	int st,ed,next;
}E[N*2];
void addedge(int x,int y)
{
	E[num].st=x;
	E[num].ed=y;
	E[num].next=head[x];
	head[x]=num++;
}
void dfs(int u)
{
	vis[u]=1;
	int i,v;
	for(i=head[u];i!=-1;i=E[i].next)
	{
		v=E[i].ed;
		if(vis[v]==1)continue;
		dfs(v);
		dp[u]+=(dp[v]+Node[v]+1);//所有子节点到到父节点的距离
		Node[u]+=(Node[v]+1);//子节点个数
	}
}
long long  mm;
void dfs1(int u)
{
	int i,v;
	vis[u]=1;
	for(i=head[u];i!=-1;i=E[i].next)
	{
		v=E[i].ed;
		if(vis[v]==1)continue;
		dp1[v]+=(dp1[u]+(dp[u]-dp[v]-Node[v]-1)+n-Node[v]-1);//除了子节点外所有节点到该点的距离
		dfs1(v);
	}
	if(mm>dp[u]+dp1[u])
		mm=dp[u]+dp1[u];
}
int main()
{
	int i,x,y,t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d%d",&n,&I,&R);
		memset(head,-1,sizeof(head));
		num=0;
		for(i=1;i<n;i++)
		{
			scanf("%d%d",&x,&y);
			addedge(x,y);
			addedge(y,x);
		}
		memset(dp,0,sizeof(dp));
		memset(dp1,0,sizeof(dp1));
		memset(Node,0,sizeof(Node));
		memset(vis,0,sizeof(vis));
		mm=inf;
		dfs(1);
		memset(vis,0,sizeof(vis));
		dfs1(1);
		printf("%lld\n",I*I*R*mm);
		for(i=1;i<=n;i++)
		{
		  if(dp[i]+dp1[i]==mm)
			  printf("%d ",i);
		}
		printf("\n\n");
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值