Codeforces Round #359 (Div. 2) D

D - Kay and Snowflake

题目大意：给你一棵数q个询问，每个询问给你一个顶点编号，要你求以这个点为根的子树的重心是哪个节点。

定义：一棵树的顶点数为n，将重心去掉了以后所有子树的顶点的个数的两倍不会超过n。

性质 1 ：树中所有点到某个点的距离和中，到重心的距离和是最小的，如果有两个距离和，他们的距离和一样。

性质 2 ：把两棵树通过某一点相连得到一颗新的树，新的树的重心必然在连接原来两棵树重心的路径上。

性质 3 ：一棵树添加或者删除一个节点，树的重心最多只移动一条边的位置。

#include<bits/stdc++.h>
#define fi first
#define se second
#define mk make_pair
#define ll long long
#define pll pair<ll,ll>
#define pii pair<int,int>
#define read(x) scanf("%d",&x)
#define sread(x) scanf("%s",x)
#define dread(x) scanf("%lf",&x)
#define lread(x) scanf("%lld",&x)
using namespace std;


const int N=1e6+7;
const int M=1e7+7;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const int mod=1e9+7;

int n,q,sum[N],fa[N],ans[N];
vector<int> edge[N];

int dfs(int u)
{
    sum[u]=1; ans[u]=u;
    for(int v:edge[u])
        sum[u]+=dfs(v);

    for(int v:edge[u])
        if(sum[v]*2>sum[u])
            ans[u]=ans[v];

    while((sum[u]-sum[ans[u]])*2>sum[u])
        ans[u]=fa[ans[u]];

    return sum[u];
}

int main()
{
    read(n); read(q);
    for(int i=2;i<=n;i++)
    {
        int pre; read(pre);
        edge[pre].push_back(i);
        fa[i]=pre;
    }
    dfs(1);
    while(q--)
    {
        int u; read(u);
        printf("%d
",ans[u]);
    }
    return 0;
}
/*

*/