用树剖搞出DFS序,这样能把一棵子树变成一个区间,一条链变成log个区间。
用线段树维护一段连续的DFS序中有几个点。共两种操作:
第一个加点,将log个区间中的总点数统计出来,然后填满,答案为∑(区间长度-原有点数),复杂度O(n(logn)^2)
第二个删点,把一段区间中的总点数统计出来作为答案,然后全部清空。复杂度O(nlogn)
因为BZOJ的时限放到了10秒,所以是可以过的。
代码:
#include<cstdio>
#include<cstring>
using namespace std;
#define rep(i,j,k) for(i=j;i<=k;++i)
#define per(i,j,k) for(i=j;i>=k;--i)
const int N=100005;
int n,m,ans;
int ne[N<<1],he[N],to[N<<1],tot;
int pre[N],son[N],dad[N],dep[N],sz[N],dfn[N],cnt;
int T[N<<2],lz[N<<2];
void add(int x,int y){
to[++tot]=y;ne[tot]=he[x];he[x]=tot;
}
void DFS1(int x){
int i,y;sz[x]=1;
for(i=he[x];i ;i=ne[i]){
dep[y=to[i]]=dep[pre[y]=x]+1;
DFS1(y);
if(sz[y]>sz[son[x]])son[x]=y;
sz[x]+=sz[y];
}
}
void DFS2(int x){
int i,y;
dfn[x]=++cnt;
dad[x]=son[pre[x]]==x?dad[pre[x]]:x;
if(son[x])DFS2(son[x]);
for(i=he[x];i;i=ne[i])if(!dfn[y=to[i]])
DFS2(y);
}
void pd(int l,int r,int num){
if(lz[num]!=-1){
T[num]=lz[num]?r-l+1:0;
if(l!=r)lz[num<<1]=lz[num<<1|1]=lz[num];
lz[num]=-1;
}
}
void up(int l,int r,int mid,int num){
pd(l,mid,num<<1);pd(mid+1,r,num<<1|1);
T[num]=T[num<<1]+T[num<<1|1];
}
int ins(int L,int R,int l,int r,int num){
pd(l,r,num);
if(L<=l&&r<=R){
lz[num]=1;
return r-l+1-T[num];
}
int mid=l+r>>1,tmp=T[num];
if(L<=mid)ins(L,R,l,mid,num<<1);
if(R>mid)ins(L,R,mid+1,r,num<<1|1);
up(l,r,mid,num);
return T[num]-tmp;
}
int del(int L,int R,int l,int r,int num){
pd(l,r,num);
if(L<=l&&r<=R){
lz[num]=0;
return T[num];
}
int mid=l+r>>1,tmp=T[num];
if(L<=mid)del(L,R,l,mid,num<<1);
if(R>mid)del(L,R,mid+1,r,num<<1|1);
up(l,r,mid,num);
return tmp-T[num];
}
int main(){
int i,x,y;char s[10];
scanf("%d",&n);
rep(i,2,n){
scanf("%d",&x);
add(1+x,i);
}
DFS1(1);DFS2(1);
scanf("%d",&m);
memset(lz,-1,sizeof lz);
while(m--){
scanf("%s%d",s,&x);++x;
if(s[0]=='i')
for(ans=0;x;x=pre[y]){
y=dad[x];
ans+=ins(dfn[y],dfn[x],1,n,1);
}
else ans=del(dfn[x],dfn[x]+sz[x]-1,1,n,1);
printf("%d\n",ans);
}
return 0;
}