cf686D（子树的重心）

cf686D
题意：求解有根树中所有子树的重心。
徐州打铁归来补题。
首先树的重心的性质如下：

1. 树的重心一定存在于直连儿子中节点个数最大的一个中，或就是本身。
2. 树的重心的所有子树中（删去该节点）最大的一颗的节点个数一定小于等于所有节点数量和的一半。
3. 重心不下移，若求解出子树的重心，则整个树的重心一定是子树重心或子树重心的父亲。。。。父亲。。。父亲。
4. 一颗树最多只有两个重心，并且这两个重心一定是相连的。

于是可以递归求出子树的重心，直连边中最大的子树大小。
递归完毕
然后在儿子子树最大的一颗中，将其子树重心上移check其最大子树是否小于等于整个树大小的一半。
并且如果要求出一颗树所有的重心，只需要找到一个重心后，再检查一下他的父亲是不是也是重心即可。

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

const int maxn=3e5+7;
struct Edge{
    int v,next;
}edge[maxn<<1];
int head[maxn],top;
void init(){top=0;memset(head,-1,sizeof(head));}
void add(int u,int v){
    edge[top].v=v;
    edge[top].next=head[u];
    head[u]=top++;
}

vector<int> Ans[maxn];
int siz[maxn];
int Son[maxn];//以x为根的子树中直连儿子的最大值;

int pre[maxn];

bool check(int x,int fa){
    return max(Son[x],siz[fa]-siz[x])*2<=siz[fa];
}

void dfs(int u,int fa){
    siz[u]=1;
    int maxx=0,idd=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==fa) continue;
        dfs(v,u);
        siz[u]+=siz[v];
        if(siz[v]>maxx){
            maxx=siz[v];
            idd=v;
        }
    }
    Son[u]=maxx;
    if(!idd){
        Ans[u].push_back(u);
        return ;
    }

    maxx=0,idd=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==fa) continue;
        if(siz[v]>maxx){ maxx=siz[v]; idd=v; }
    }
    idd=Ans[idd][0];
    while(!check(idd,u)) idd=pre[idd];
    Ans[u].push_back(idd);
    if(idd!=u&&check(pre[idd],u)) Ans[u].push_back(pre[idd]);//第二个重心;
}

int main(){
    int n,q,u,v;
    scanf("%d%d",&n,&q);
    init();
    for(int i=2;i<=n;++i){
        scanf("%d",&u);
        v=i;
        pre[v]=u;
        add(u,v),add(v,u);
    }
    dfs(1,0);
    for(int i=1;i<=q;++i){
        scanf("%d",&u);
        printf("%d\n",Ans[u][0]);
    }

    return 0;
}