【codeforces685B686D】【Kay and Snowflake】【线段树合并】

题目大意

给定30万个点的树，要求支持30万个询问，询问以某个点为根的子树的重心。

题解

给每的节点建立权值线段树，记录每个子节点为根的子树大小。重心为根的子树大小最接近但大于全树大小的一半。使用线段树合并即可解决本问题。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=300000;
int cntgra,cntpon,n,q,to[maxn+10],next[maxn+10],begin[maxn+10],tsize[maxn*18*2+10],son[maxn*18*2+10][2],
pon[maxn*18*2+10],ans[maxn+10],size[maxn+10];
void insert(int u,int v){
to[++cntgra]=v;
next[cntgra]=begin[u];
begin[u]=cntgra;
}
int merge(int u,int v){
if(!tsize[u])return v;
if(!tsize[v])return u;
son[u][0]=merge(son[u][0],son[v][0]);
son[u][1]=merge(son[u][1],son[v][1]);
tsize[u]=tsize[son[u][0]]+tsize[son[u][1]];
return u;
}
int ask(int now,int l,int r,int tg){
int m=(l+r)/2;
if(!tsize[now])return 0;
if(l==r)return pon[now];
if(tg<=m){
int tmp=ask(son[now][0],l,m,tg);
if(tmp)return tmp;
else return ask(son[now][1],m+1,r,tg);
}
else return ask(son[now][1],m+1,r,tg);
}
void change(int now,int l,int r,int tg,int v){
int m=(l+r)/2;
tsize[now]++;
if(l==r)pon[now]=v;
else if(tg<=m){
if(!son[now][0])son[now][0]=++cntpon;
change(son[now][0],l,m,tg,v);
}
else{
if(!son[now][1])son[now][1]=++cntpon;
change(son[now][1],m+1,r,tg,v);
}
}
void dfs(int now){
size[now]=1;
for(int i=begin[now];i!=-1;i=next[i]){
dfs(to[i]);
size[now]+=size[to[i]];
son[now][0]=merge(son[now][0],son[to[i]][0]);
son[now][1]=merge(son[now][1],son[to[i]][1]);
}
change(now,1,n,size[now],now);
ans[now]=ask(now,1,n,(size[now]+1)/2);
}
int main(){
scanf("%d%d",&n,&q);
memset(begin,255,sizeof(begin));cntpon=n;
fo(i,2,n){
int x;scanf("%d",&x);
insert(x,i);
}
dfs(1);
fo(i,1,q){
int x;scanf("%d",&x);
printf("%d\n",ans[x]);
}
return 0;
}