AcWing 846. 树的重心
原题链接
简单
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
const int N = 1e5+10,M=N*2;
int e[M],ne[M],h[N],idx;
bool st[N];
int ans=N;
void add(int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u)//返回以u为根节点的子树节点的数量
{
int sum=1,res=0;
st[u]=true;
for(int i=h[u];~i;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()
{
cin>>n;
memset(h,-1,sizeof h);
for(int i=0;i<n-1;i++)
{
int a,b;
cin>>a>>b;
add(a,b);
add(b,a);
}
dfs(1);
cout<<ans;
return 0;
}