D. Kay and Snowflake

题目链接:http://codeforces.com/contest/686/problem/D

题意:

给你n个点,以1为根,问你每一颗子树的重心是哪一个节点。

解题思路:

很明显的树形DP,每棵树的重心一定在每个子树的重心到该点的路径上,但是如果考虑这些路径所有的点可能会超时,这是我们可以想到:如果该树的子树的结点数大于该树的两倍那么就不用往上找了。

#include<bits/stdc++.h>
using namespace std;
const int maxn=300005;
const int inf=0x3f3f3f3f;
struct st{
	int to,next;
}stm[maxn*2];
int head[maxn];
int cnt;
void add(int u,int v){
	stm[cnt].to=v;
	stm[cnt].next=head[u];
	head[u]=cnt++;
}
int ans[maxn];
int dp[maxn];//最大子树大小 
int dp1[maxn];//当前树大小 
int pre[maxn];
void dfs(int now,int fa){
	dp1[now]=1;
	dp[now]=0;
	ans[now]=now;
	for(int i=head[now];~i;i=stm[i].next){
		int to=stm[i].to;
	
		if(to==fa)continue;
		pre[to]=now;
		dfs(to,now);
		dp1[now]+=dp1[to];
		if(dp[now]<=dp1[to]){
			dp[now]=dp1[to];
		}
	}
	int tem=dp1[now];
	for(int i=head[now];~i;i=stm[i].next){
		int pos=stm[i].to;
		if(pos==fa)continue;
		
		int tem1=ans[pos];
		while(dp1[tem1]*2<=dp1[now]){
			if(tem>=max(dp[tem1],dp1[now]-dp1[tem1])){
				ans[now]=tem1;
				tem=max(dp[tem1],dp1[now]-dp1[tem1]);
			}	
			tem1=pre[tem1];
		}
		if(tem>=max(dp[tem1],dp1[now]-dp1[tem1])){
			ans[now]=tem1;
			tem=max(dp[tem1],dp1[now]-dp1[tem1]);
		}
	}
		
}
int main(){
	int n,q;
	int tem;
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&q);
	pre[1]=1;
	for(int i=2;i<=n;i++){
		pre[i]=i;
		scanf("%d",&tem);
		add(i,tem);
		add(tem,i);
	}
	dfs(1,0);
	//for(int i=1;i<=n;i++)cout<<dp1[i]<<endl;
	for(int i=1;i<=q;i++){
		scanf("%d",&tem);
		printf("%d\n",ans[tem]);
	}
	return 0;
} 

  

posted @ 2019-11-07 20:19  风雨兼程-zhi  阅读(146)  评论(0编辑  收藏  举报