树的dfs(AcWing树的重心)

注释很详细了

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e6+10,M=2*N;
typedef long long ll;
int h[N],e[M],ne[M],idx;
bool st[M];
int ans=N;
int n;
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
//返回以u为根的子树的点的大小
int dfs(int u)
{
	st[u]=true;//已经被搜过了
	int sum=1,res=0;
	for(int i=h[u];i!=-1;i=ne[i])//只需要一次深度遍历
	{
		int j=e[i];//该节点所在(链表)树的位置
		if(!st[j])
		{
			int s=dfs(j);
			res=max(res,s);//该节点下最大连通块点数
			sum+=s;//以该节点为根的连通块的点数
		}
	}
	res=max(res,n-sum);//与该节点之外的连通块点数比较
	ans=min(ans,res);//边遍历节点边记录最小的节点个数
	return sum;
}
int main()
{
	IOS
	cin>>n;
	memset(h,-1,sizeof h);
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		add(a,b),add(b,a);
	}
	dfs(1);//只需要遍历一次,可以从任意点遍历
	cout<<ans<<endl;
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值