codeforces685B. Kay and Snowflake(树形dp)

传送门
题意简述:给出一棵树,求每个子树的重心。


首先通过画图可以观察出一个性质,我们从叶子结点向根节点递推重心的话重心的位置是不会下降的。
然后由于一个点的重心要么是自己,要么在重儿子子树内,因此如果重心不是自己,我们从重儿子子树的重心向自己爬统计答案就行了,由于重心不会下降因此时间复杂度 O ( n ) O(n) O(n)
代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
const int N=3e5+5;
int n,q,siz[N],fa[N],dep[N],hson[N],ans[N];
vector<int>e[N];
void dfs1(int p){
	siz[p]=1;
	for(ri i=0,v;i<e[p].size();++i){
		v=e[p][i];
		fa[v]=p,dep[v]=dep[p]+1,dfs1(v),siz[p]+=siz[v];
		if(siz[v]>siz[hson[p]])hson[p]=v;
	}
}
void dfs2(int p){
	if(!hson[p]){ans[p]=p;return;}
	for(ri i=0;i<e[p].size();++i)dfs2(e[p][i]);
	if(siz[hson[p]]*2<=siz[p])ans[p]=p;
	else for(ri v=ans[hson[p]];v!=p;v=fa[v])if(max(siz[hson[v]],siz[p]-siz[v])*2<=siz[p]){ans[p]=v;break;}
}
int main(){
	n=read(),q=read();
	for(ri i=2;i<=n;++i)e[read()].push_back(i);
	dfs1(1),dfs2(1);
	while(q--)cout<<ans[read()]<<'\n';
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值