Kay and Snowflake CodeForces - 686D

原题链接
考察:树形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;
}