【NOI2015】软件包管理器

【NOI2015】软件包管理器

树链剖分的第二题，虽然抄了板子题不少，但还算简单吧。
【注意是从$0$开始的哦】

线段树和树链剖分的基本操作啦，就先这样吧。

树链剖分入门

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
struct edge{int v,nxt;}e[N];
int first[N],cnt=0;
inline void add(int u,int v){e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;}
int n,m,fa[N],dep[N],siz[N],son[N],top[N],pos[N],idx[N],end[N],tot;
char s[15];
void dfs1(int x){
	siz[x]=1;
	for(int i=first[x];i;i=e[i].nxt){
		int v=e[i].v;
		dep[v]=dep[x]+1;
		dfs1(v);
		siz[x]+=siz[v];
		if(siz[v]>siz[son[x]])son[x]=v;
	}
}
void dfs2(int x){
	if(son[x]){
		pos[son[x]]=++tot;
		idx[tot]=son[x];
		top[son[x]]=top[x];
		dfs2(son[x]);
	}
	for(int i=first[x];i;i=e[i].nxt){
		int v=e[i].v;
		if(v==son[x])continue;
		pos[v]=++tot;
		idx[tot]=v;
		top[v]=v;
		dfs2(v);
	}
	end[x]=tot;
}
#define lc (p<<1)
#define rc (p<<1|1)
struct tree{
	int l,r,sum,lazy;//l,r可以不存,直接带入 
}t[N<<2];
void build(int p,int l,int r){
	t[p].l=l;t[p].r=r;t[p].sum=0;t[p].lazy=-1;
	if(l==r)return;
	int mid=(l+r)>>1;
	build(lc,l,mid);build(rc,mid+1,r);
}
void init(){
	dfs1(1);
	top[1]=idx[1]=pos[1]=tot=1;
	dfs2(1);
	build(1,1,n);
}
void pushdown(int p){
	if(t[p].lazy==-1)return;
	t[lc].sum=(t[lc].r-t[lc].l+1)*t[p].lazy;
	t[lc].lazy=t[p].lazy;
	t[rc].sum=(t[rc].r-t[rc].l+1)*t[p].lazy;
	t[rc].lazy=t[p].lazy;
	t[p].lazy=-1;
} 
int query(int p,int l,int r){
	if(l<=t[p].l&&t[p].r<=r)return t[p].sum;
	pushdown(p);
	int mid=(t[p].l+t[p].r)>>1,ans=0;
	if(l<=mid)ans+=query(lc,l,r);
	if(mid<r)ans+=query(rc,l,r);
	return ans;
}
void change(int p,int l,int r,int val){
	if(l<=t[p].l&&t[p].r<=r){
		t[p].lazy=val;
		t[p].sum=(t[p].r-t[p].l+1)*t[p].lazy;
		return;
	}
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid)change(lc,l,r,val);
	if(mid<r)change(rc,l,r,val);
	t[p].sum=t[lc].sum+t[rc].sum;
}
int solve1(int x){
	int y=1,ans=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		ans+=pos[x]-pos[top[x]]+1-query(1,pos[top[x]],pos[x]);
		change(1,pos[top[x]],pos[x],1);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])swap(x,y);
	ans+=pos[x]-pos[y]+1-query(1,pos[y],pos[x]);
	change(1,pos[y],pos[x],1);
	return ans;
}
int solve2(int x){
	int ans=query(1,pos[x],end[x]);
	change(1,pos[x],end[x],0);
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=2;i<=n;i++){
		scanf("%d",&fa[i]);fa[i]++;
		add(fa[i],i);
	}
	scanf("%d",&m);
	init();
	int x;
	while(m--){
		scanf("%s",s+1);//遇到空格结束
		scanf("%d",&x);
		if(s[1]=='i')printf("%d\n",solve1(x+1));//安装
		else printf("%d\n",solve2(x+1));//卸载 
	}
	return 0;
}