E32 树形DP 树的重心

视频链接：E32 树形DP 树的重心_哔哩哔哩_bilibili

 

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N=100010;
vector<int> e[N];
int n,sz[N],ans=1e9;

void dfs(int u,int fa){
  sz[u]=1; int mx=0;
  for(auto v:e[u]){
    if(v==fa) continue;
    dfs(v,u);
    sz[u]+=sz[v];
    mx=max(mx,sz[v]);
  }
  mx=max(mx,n-sz[u]);
  ans=min(ans,mx);
}
int main(){
  scanf("%d",&n);
  for(int i=1,a,b;i<n;i++){
    scanf("%d%d",&a,&b);
    e[a].push_back(b);
    e[b].push_back(a);
  }
  dfs(1,0);
  printf("%d\n",ans);
}

 

练习：

1655 -- Balancing Act (poj.org)

Kay and Snowflake - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

P1364 医院设置 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)