[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
=y;
adde(x,n),adde(n,x),f
=x;
if(a[bl[x]].size==block){
bl
=++cnt;a[cnt].insert(y);
add(bl[x],cnt);
}else a[bl
=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;
}