内容摘自《算法竞赛进阶指南》第三版,作者李煜东
每个节点x为根的子树大小为size[x]。对于叶子节点,我们已知“以它为根的子树”大小为1。若结点x有k个子节点,y1~yk,并且以y1 ~yk为根的子数大小为sizey1,sizey2,…,size yk,则以x为根的子树大小就是sizex=sizey1+sizey2+…+sizeyk+1;
对于一个节点x,如果我们把他从树中删除,那么原来的一棵树就会分成若干不相连的部分,其中每一部分都是一棵子树。设max_part(x)表示在删除节点x后产生的子数中,最大的一棵的大小。是max_part函数取得最小值的节点p就称为整棵树的重心。
通过下面的代码我们可以统计出size数组,并求出树的重心
void dfs(){
v[x]=1;size[x]=1;
int max_part=0;
for(int i=head[x];i;i=next[i]){
int y=ver[i];
if(v[y])continue;//点y已经被访问过了
dfs(y);
size[x]=+=size[y];
max_part=max(max_part,size[y]);
}
max_part=max(max_part,n-size[x]);//n为整棵树的节点数目
if(max_part<ans){
ans=max_part;//ans记录重心对应的max_part值
pos=x;//pos记录重心的位置
}