4196: [Noi2015]软件包管理器

用树剖搞出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;
} 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值