原题链接
考察:树形dp(?)
思路:
主要利用的两个性质:
(1) 以树的重心为根时,所有子树的大小都不超过整棵树大小的一半.
(2) 把两棵树通过一条边相连得到一棵新的树,那么新的树的重心在连接原来两棵树的重心的路径上.
注意性质1不是子树的和,以贪心来考虑的话,我们对于根节点\(u\),其最大的子树根为\(v\),那么\(sz[v]*2<=sz[u]\),则说明\(u\)是重心.
因此,对于每个点我们记录最大的子节点\(son[u]\),子节点个数\(sz[u]\).
Code
#include <iostream>
#include <cstring>
using namespace std;
const int N = 300010;
int h[N],idx,f[N],n,Q,sz[N],fa[N],ans[N],son[N];
struct Road{
int to,ne;
}road[N<<1];
void add(int a,int b)
{
road[idx].to = b,road[idx].ne = h[a],h[a] = idx++;
}
int dfs(int u)
{
int maxn = 0,tar = 0;
sz[u] = 1;
for(int i=h[u];~i;i=road[i].ne)
{
int v = road[i].to;
sz[u]+=dfs(v);
if(sz[v]>sz[son[u]]) son[u] = v;
}
if(sz[son[u]]*2<=sz[u]) ans[u] = u;
else{
int t = ans[son[u]];
while(fa[t]&&max(sz[son[t]],sz[u]-sz[t])*2>sz[u]) t = fa[t];
ans[u] = t;
}
return sz[u];
}
int main()
{
scanf("%d%d",&n,&Q);
for(int i=1;i<=n;i++) h[i] = -1,sz[i] = 1;
for(int i=2,j;i<=n;i++)
{
scanf("%d",&j);
add(j,i);
fa[i] = j;
}
dfs(1);
while(Q--)
{
int x;
scanf("%d",&x);
printf("%d\n",ans[x]);
}
return 0;
}