[BZOJ 3720]Gty的妹子樹 樹上分塊

類似於序列上面的分塊,給每一個樹設置一個大小上限,不夠就繼續加,夠了就新建一個塊.

友情提示:把塊寫成一個結構體打包起來,然後數組開大一點,血的教訓

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define maxn 100210
using namespace std;
struct blo{
	int a[251],size;
	int kth(int x){return size-(upper_bound(a+1,a+size+1,x)-a-1);}
	void init(int x){a[++size]=x;}
	void sor(){sort(a+1,a+1+size);}
	void insert(int x){
		int i;size++;
		for(i=size;i>=1&&a[i-1]>x;i--)a[i]=a[i-1];
		a[i]=x;
	}
	void update(int x,int y){
		int i=lower_bound(a+1,a+1+size,x)-a;
		for(;i<size&&a[i+1]<y;i++)a[i]=a[i+1];
		for(;i>1&&a[i-1]>y;i--)a[i]=a[i-1];
		a[i]=y;
	}
}a[10000];
int n,m,head[maxn],last[maxn],tot=1,cc=1,w[maxn],bl[maxn];
int cnt,block,ans,f[maxn];
struct edge{int v,next;}e[maxn*2],ed[maxn*2];
void adde(int a,int b){e[tot].v=b,e[tot].next=head[a];head[a]=tot++;}
void add(int a,int b){ed[cc].v=b,ed[cc].next=last[a];last[a]=cc++;}
void dfs(int u,int fa){
	f[u]=fa;
	if(a[bl[fa]].size==block||u==1){
		bl[u]=++cnt,a[cnt].init(w[u]);
		if(u!=1)add(bl[fa],cnt);
	}else a[bl[u]=bl[fa]].init(w[u]);
	for(int v,i=head[u];i;i=e[i].next){
		if((v=e[i].v)==fa)continue;
		dfs(v,u);
	}
}
void query(int u,int x){
	ans+=a[u].kth(x);
	for(int i=last[u];i;i=ed[i].next)query(ed[i].v,x);
}
void get(int u,int x){
	ans+=(w[u]>x);
	for(int v,i=head[u];i;i=e[i].next){
		if((v=e[i].v)==f[u])continue;
		if(bl[u]==bl[v])get(v,x);
		else query(bl[v],x);
	}
}
void solve(){
	int m,pos,x,y;
	scanf("%d",&m);
	while(m--){
		scanf("%d%d%d",&pos,&x,&y);
		x^=ans,y^=ans;
		if(pos==0){
			ans=0;get(x,y);
			printf("%d\n",ans);
		}else if(pos==1){
			a[bl[x]].update(w[x],y);
			w[x]=y;
		}else{
			n++;w[n]=y;
			adde(x,n),adde(n,x),f[n]=x;
			if(a[bl[x]].size==block){
				bl[n]=++cnt;a[cnt].insert(y);
				add(bl[x],cnt);
			}else a[bl[n]=bl[x]].insert(y);
		}
	}
}
int main(){
	scanf("%d",&n);block=sqrt(n);
	for(int x,y,i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		adde(x,y),adde(y,x);
	}
	for(int i=1;i<=n;i++)scanf("%d",w+i);
	dfs(1,0);
	for(int i=1;i<=cnt;i++)a[i].sor();
	solve();
	return 0;
}