树分块
每S个节点分一块,如果x的父亲所属块超过S就自己分一块。
块内的所有权值排序。
对于询问,块内二分,块外暴力。
对于修改和加入,维护排序就好。
#include<cstdio>
#include<vector>
#include<algorithm>
#define N 60005
#define S 472//经过计算 这个值是最佳的
using namespace std;
struct edge{int next,to;}e[N*10];
int w[N], last[N], last2[N], ecnt=1, belong[N], bcnt, top[N], siz[N], q[N], fa[N], n;
vector<int> data[N];
void add1(int a, int b)
{
e[++ecnt]=(edge){last[a],b};
last[a]=ecnt;
}
void add2(int a, int b)
{
e[++ecnt]=(edge){last2[a],b};
last2[a]=ecnt;
}
void newnode(int x, int f)
{
if(siz[belong[f]]<S)
{
belong[x]=belong[f];
}
else
{
belong[x]=++bcnt;
top[belong[x]]=x;
add2(top[belong[f]],x);
}
++siz[belong[x]];
data[belong[x]].push_back(w[x]);
}
void bfs()
{
belong[0]=bcnt=1;
top[bcnt]=1;
q[0]=1;
for(int head=0, tail=1; head<tail; head++)
{
int x=q[head], f=fa[x];
newnode(x,f);
for(int i = last[x]; i; i = e[i].next)
{
int y=e[i].to;
if(belong[y])continue;
fa[y]=x;
q[tail++]=y;
}
}
}
int solve(int x, int val)
{
int l=-1, r=siz[x]-1;
while(l<r)
{
int mid=(l+r+1)>>1;
if(data[x][mid]<=val)l=mid;
else r=mid-1;
}
return siz[x]-l-1;
}
int query(int x, int val)
{
int ret=0;
if(top[belong[x]]==x)
{
ret+=solve(belong[x], val);
for(int i = last2[x]; i; i=e[i].next)
ret+=query(e[i].to,val);
}
else
{
ret+=(w[x]>val);
for(int i = last[x]; i; i=e[i].next)
{
if(fa[e[i].to]==x)
ret+=query(e[i].to,val);
}
}
return ret;
}
void modify(int x, int val)
{
int block=belong[x], pos;
for(pos = 0; pos < siz[block]; pos++)
if(data[block][pos]==w[x])break;
data[block][pos]=w[x]=val;
while(pos < siz[block]-1 && data[block][pos]>data[block][pos+1])
{
swap(data[block][pos], data[block][pos+1]);
pos++;
}
while(pos > 0 && data[block][pos]<data[block][pos-1])
{
swap(data[block][pos], data[block][pos-1]);
pos--;
}
}
void add(int f, int val)
{
w[++n]=val;
add1(f,n);
fa[n]=f;
newnode(n,f);
for(int pos=siz[belong[n]]-1; pos>0; pos--)
{
if(data[belong[n]][pos]<data[belong[n]][pos-1])
swap(data[belong[n]][pos],data[belong[n]][pos-1]);
else break;
}
}
int main()
{
scanf("%d",&n);
for(int i = 1, u, v; i < n; i++)
{
scanf("%d%d",&u,&v);
add1(u,v);
add1(v,u);
}
for(int i = 1; i <= n; i++)
scanf("%d",&w[i]);
bfs();
for(int i = 1; i <= bcnt; i++)
sort(data[i].begin(), data[i].end());
int m;
scanf("%d", &m);
int ans=0;
while(m--)
{
int op, u, v;
scanf("%d%d%d", &op, &u, &v);
u^=ans; v^=ans;
if(op==0)
{
ans=query(u,v);
printf("%d\n",ans);
}
else if(op==1)
modify(u,v);
else
add(u,v);
}
}