将树分块,保证每块内都联通,然后保证每块内都有序,就可以支持查询操作了。
加入节点的时候,如果达到块的大小就重建一个块,否则加入当前块,不要忘记加入后对块排序!
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=60005;
int head[N],list[N<<1],next[N<<1];
int HEAD[N],LIST[N<<1],NEXT[N<<1];
int a[N],fa[N],belong[N];
int n,m,cnt,ans,CNT,tot,B;
struct node
{
int size;
int a[205];
inline void insert(int x)
{
a[++size]=x;
}
inline void change(int x,int y)
{
int t=lower_bound(a+1,a+size+1,x)-a;
a[t]=y;
sort(a+1,a+size+1);
}
inline int query(int x)
{
int t=upper_bound(a+1,a+size+1,x)-a;
return size-t+1;
}
};
node block[10005];
inline int read()
{
int a=0,f=1; char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
return a*f;
}
inline void insert(int x,int y)
{
next[++cnt]=head[x];
head[x]=cnt;
list[cnt]=y;
}
inline void INSERT(int x,int y)
{
NEXT[++CNT]=HEAD[x];
HEAD[x]=CNT;
LIST[CNT]=y;
}
void dfs(int x)
{
if (block[belong[fa[x]]].size==B)
belong[x]=++tot,block[tot].insert(a[x]),INSERT(belong[fa[x]],tot);
else belong[x]=belong[fa[x]],block[belong[x]].insert(a[x]);
for (int i=head[x];i;i=next[i])
if (list[i]!=fa[x])
fa[list[i]]=x,dfs(list[i]);
}
void dfs1(int x,int y)
{
ans+=block[x].query(y);
for (int i=HEAD[x];i;i=NEXT[i])
dfs1(LIST[i],y);
}
void query(int x,int y)
{
if (a[x]>y) ans++;
for (int i=head[x];i;i=next[i])
if (list[i]!=fa[x])
{
if (belong[list[i]]==belong[x]) query(list[i],y);
else dfs1(belong[list[i]],y);
}
}
int main()
{
n=read(); B=sqrt(n);
for (int i=1;i<n;i++)
{
int u=read(),v=read();
insert(u,v); insert(v,u);
}
for (int i=1;i<=n;i++) a[i]=read();
dfs(1);
for (int i=0;i<=tot;i++)
sort(block[i].a+1,block[i].a+block[i].size+1);
m=read();
for (int i=1;i<=m;i++)
{
int opt=read(),u=read()^ans,x=read()^ans;
if (opt==0)
{
ans=0;
query(u,x);
printf("%d\n",ans);
}
else if (opt==1)
{
block[belong[u]].change(a[u],x);
a[u]=x;
}
else
{
a[++n]=x;
insert(u,n);
fa[n]=u;
if (block[belong[u]].size==B)
belong[n]=++tot,block[tot].insert(x),INSERT(belong[u],tot);
else
{
belong[n]=belong[u],block[belong[n]].insert(a[n]);
sort(block[belong[n]].a+1,block[belong[n]].a+block[belong[n]].size+1);
}
}
}
return 0;
}