CF685B Kay and Snowflake

链接

CF

题目大意

给你一颗树，询问子树的重心

思路

贪心？
重心肯定是向上走的，所以直接向上跳就好了。
不优秀的时候就不要跳了 ，因为以后也不能更新了。
复杂度O（n）（没大仔细想过）

代码

#include <bits/stdc++.h>
using namespace std;
const int N=3e5+7;
int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
vector<int> G[N];
int n,m,siz[N],ans[N],fa[N],ma[N];
void dfs_siz(int u) {
    siz[u]++;
    for(vector<int>::iterator v=G[u].begin();v!=G[u].end();++v) {
        dfs_siz(*v);
        siz[u]+=siz[*v];    
        ma[u]=max(ma[u],siz[*v]);
    }
}
void dfs(int u) {
    ans[u]=u;
    for(vector<int>::iterator v=G[u].begin();v!=G[u].end();++v) {
        dfs(*v);
        int tmp=ans[*v],now=ma[u];
        while(tmp!=u) {
            if(max(ma[tmp],siz[u]-siz[tmp]) < now) {
                now=max(ma[tmp],siz[u]-siz[tmp]);
                ans[u]=tmp;
            }
            if(ma[tmp]>siz[u]-siz[tmp]) break;
            tmp=fa[tmp];
        }
    }
}
int main() {
    // freopen("a.in","r",stdin);
    n=read(),m=read();
    for(int i=2;i<=n;++i) {
        fa[i]=read();
        G[fa[i]].push_back(i);
    }
    dfs_siz(1);
    dfs(1);
    for(int i=1;i<=m;++i) printf("%d\n",ans[read()]);
    return 0;
}