动态规划特训:树的重心(树形dp)

解题思路:选择一个节点作为根,设dp[i]表示以i为根的树的总节点个数,dp[i]=满足j为其子节点的的d[j]之和再加1(根节点)。只需在dfs过程中找到最大的子树节点,并与其上方的节点数做比较,就可以找出树的重心了。

题目大意:对于一棵无根树,找到一个点使得树以该点为根的有根树,最大子树(选择该节点后其子树的最大节点)的节点数最小。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define inf 1<<30

int n,dp[101],node,minans;    
vector<int> edge[101];

int dfs(int m,int fa)
{
	int maxnode=0;
	for(int i=0;i<edge[m].size();i++)
	{
		int tmp=edge[m][i];
		if(tmp!=fa) 
		{
			dp[m]+=dfs(tmp,m);
			maxnode=max(maxnode,dp[tmp]);   //得到最大子树节点数 
		}
	}
	maxnode=max(maxnode,n-dp[m]);   //再与上方的节点数做比较 
	if(maxnode<minans)
	{
		minans=maxnode;
		node=m;
	}
	return dp[m];
}

int main()
{
	while(cin>>n)
	{
		for(int i=1;i<=n;i++) edge[i].clear();
		int a,b;
		for(int i=1;i<=n-1;i++)
		{
			cin>>a>>b;
			edge[a].push_back(b);
			edge[b].push_back(a);
		}
		for(int i=1;i<=n;i++)    //初始化都是1,即根节点本身 
		{
			dp[i]=1;
		}
		minans=inf;
		dfs(1,-1);
		cout<<minans<<" "<<node<<endl;
	}
	return 0;
} 

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值