AcWing 846. 树的重心【树形DP】
原题链接
简单
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1e5 + 10, M = N * 2;
int e[M], ne[M], h[N], idx;
bool st[N];
int ans = N;
int n;
void add(int a,int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int dfs(int u)
{
st[u] = true;
int sum = 1; // 当前这是一个点
int 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); // res表示删除该结点后所有联通块的最大结点数,而ans记录其中最小的值(每遍历完一个点后都要进行的操作)
return sum;
}
int main()
{
memset(h,-1,sizeof h);
cin >> n;
for(int i = 0;i < n;i++)
{
int a, b;
cin >> a >> b;
add(a,b), add(b,a);
}
dfs(1);
cout << ans;
return 0;
}