树剖 10pts 求助
查看原帖
树剖 10pts 求助
356925
快斗游鹿楼主2022/11/13 17:42

RT,只过了两个点。

/*
思路:安装软件答案就是节点深度-到根节点1的个数,然后将整个区间变为1。卸载软件答案就是子树内1的个数,然后将整个区间变为0。
*/
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
struct Node{
	int l,r;
	long long sum,zero,one;//l,r左端点,右端点,sum为区间和,zero表示当前区间是否要全改为0,one表示是否要全改为1
}t[N*4];
struct edge{
	int to,nxt;
}e[N*2];
int n,cnt,head[N];
int deep[N],fa[N],size[N],son[N];
int id[N],top[N];
int read(){
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		if(ch=='-')f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=x*10+ch-'0';
		ch=getchar();
	}
	return x*f;
}
void add(int u,int v){//链式前向星加边
	cnt++;
	e[cnt].to=v;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs1(int u,int f,int d){//预处理1
	deep[u]=d;
	fa[u]=f;
	size[u]=1;
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=f){
			dfs1(v,u,d+1);
			size[u]+=size[v];
			if(size[v]>size[son[u]])son[u]=v;
		}
	}
} 
void dfs2(int u,int f){//预处理2
	cnt++;
	id[u]=cnt;
	top[u]=f;
	if(!son[u])return;
	dfs2(son[u],f);
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v!=son[u]&&v!=fa[u]){
			dfs2(v,v);
		}
	}
}
void spread(int p){//懒标记下传
	if(t[p].one){
		t[p<<1].sum=t[p<<1].r-t[p<<1].l+1;
		t[p<<1|1].sum=t[p<<1|1].r-t[p<<1|1].l+1;
		t[p<<1].one=1;t[p<<1|1].one=1;t[p].one=0;
	}
	if(t[p].zero){
		t[p<<1].sum=0;
		t[p<<1|1].sum=0;
		t[p<<1].zero=1;t[p<<1|1].zero=1;t[p].zero=0;
	} 
}
void build(int p,int l,int r){//建树
	t[p].l=l;t[p].r=r;
	if(l==r){
		return;
	}
	int mid=(l+r)>>1;
	build(p<<1,l,mid);
	build(p<<1|1,mid+1,r);
}
void change(int p,int l,int r,int z){//修改,z=1表示整个区间改为1,反之表示改为0
	if(l<=t[p].l&&t[p].r<=r){
		if(z)t[p].one=1,t[p].sum=t[p].r-t[p].l+1;
		else t[p].zero=1,t[p].sum=0;;
		return;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid)change(p<<1,l,r,z);
	if(r>mid)change(p<<1|1,l,r,z);
	t[p].sum=t[p<<1].sum+t[p<<1|1].sum;
}
long long ask(int p,int l,int r){//询问
	if(l<=t[p].l&&t[p].r<=r){
		return t[p].sum;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	long long res=0;
	if(l<=mid)res+=ask(p<<1,l,r);
	if(r>mid)res+=ask(p<<1|1,l,r);
	return res;
}
long long solve1(int x){//针对安装操作
	long long ans=0;
	while(top[x]!=1){
		ans+=ask(1,id[top[x]],id[x]);
		change(1,id[top[x]],id[x],1);
		x=fa[top[x]];
	}
	ans+=ask(1,1,id[x]);
	change(1,1,id[x],1);
	return ans;
}
int main(){
	//freopen("P2146_1.in","r",stdin);
	//freopen("std.out","w",stdout);
	n=read();
	for(int i=2;i<=n;i++){
		int v;v=read();v++;
		add(i,v);add(v,i);
	}
	dfs1(1,0,1);
	//for(int i=1;i<=n;i++)cout<<i<<" "<<"deep:"<<deep[i]<<" "<<"fa:"<<fa[i]<<" "<<"size:"<<size[i]<<" "<<"son:"<<son[i]<<endl;
	cnt=0;dfs2(1,1);
	//for(int i=1;i<=n;i++)cout<<i<<" "<<"id:"<<id[i]<<" "<<"top:"<<top[i]<<endl;
	build(1,1,n);
	int q;q=read();
	while(q--){
		string s;cin>>s;int x;x=read();x++;
		if(s=="install")printf("%lld\n",deep[x]-solve1(x));
		else{
			printf("%lld\n",ask(1,id[x],id[x]+size[x]-1));
			change(1,id[x],id[x]+size[x]-1,0);
		}
	}
	return 0;
}

2022/11/13 17:42
加载中...