###Description
一棵 n 个点的树, 树根为 1. 一开始每个点上有一个 1…n 的颜色 ci, 不同点颜色可以相同.
现在有 q 次操作, 分为两种类型:
- 1 u l r: 询问子树 u 中有多少种在 l 到 r 之间的颜色至少出现了一次
- 2 u c: 将 u 的颜色修改为 c
强制在线
###Solution
如果只考虑询问一个子树的颜色总数,对于每种颜色,点按dfs序排序,每个点上+1,相邻点lca-1,于是就是经典的子树求和。考虑修改和查询,直接树状数组套线段树即可。
###Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<set>
#define fo(i,j,k) for(int i=j;i<=k;++i)
#define fd(i,j,k) for(int i=j;i>=k;--i)
#define rep(i,x) for(int i=ls[x];i;i=nx[i])
using namespace std;
const int N=1e5+10;
int to[N<<1],nx[N<<1],ls[N],num=0;
int fa[N][17],dep[N],fr[N],en[N],z[N];
multiset<int> cl[N];
typedef multiset<int> :: iterator it;
struct node{
int l,r,s;
}tr[N<<8];
int rt[N],n,dfn=0,tot=0;
void link(int u,int v){
to[++num]=v,nx[num]=ls[u],ls[u]=num;
}
int col[N];
void pre(int x){
fr[x]=++dfn,z[dfn]=x;
fo(i,1,16) fa[x][i]=fa[fa[x][i-1]][i-1];
rep(i,x){
int v=to[i];
if(v==fa[x][0]) continue;
fa[v][0]=x,dep[v]=dep[x]+1;
pre(v);
}
en[x]=dfn;
}
int lca(int u,int v){
if(dep[u]<dep[v]) swap(u,v);
fd(i,16,0) if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];
if(u==v) return u;
fd(i,16,0) if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
return fa[u][0];
}
int seg_add(int v,int l,int r,int x,int t){
if(!v) v=++tot;
tr[v].s+=t;
if(l==r) return v;
int mid=(l+r)>>1;
x<=mid?tr[v].l=seg_add(tr[v].l,l,mid,x,t):tr[v].r=seg_add(tr[v].r,mid+1,r,x,t);
return v;
}
int seg_sum(int v,int l,int r,int x,int y){
if(!v) return 0;
if(l==x && r==y) return tr[v].s;
int mid=(l+r)>>1;
if(y<=mid) return seg_sum(tr[v].l,l,mid,x,y);
else if(x>mid) return seg_sum(tr[v].r,mid+1,r,x,y);
else return seg_sum(tr[v].l,l,mid,x,mid)+seg_sum(tr[v].r,mid+1,r,mid+1,y);
}
void add(int x,int p,int t){
for(;x<=n;x+=x&-x) rt[x]=seg_add(rt[x],1,n,fr[p],t);
}
int sum(int x,int p){
int tmp=0;
for(;x;x-=x&-x) tmp+=seg_sum(rt[x],1,n,fr[p],en[p]);
return tmp;
}
void modify(int x,int t){
add(col[x],x,t);
it pos=cl[col[x]].find(fr[x]);
it p=pos,q=pos;p--,q++;
if(*p && *q<=n) add(col[x],lca(z[*p],z[*q]),t);
if(*q<=n) add(col[x],lca(z[*pos],z[*q]),-t);
if(*p) add(col[x],lca(z[*pos],z[*p]),-t);
}
int main()
{
freopen("xmastree1.in","r",stdin);
freopen("xmastree1.out","w",stdout);
int q,t;
scanf("%d %d %d",&n,&q,&t);
fo(i,1,n) scanf("%d",&col[i]);
fo(i,2,n){
int u,v;
scanf("%d %d",&u,&v);
link(u,v),link(v,u);
}
dep[1]=1,pre(1);
fo(i,1,n) cl[i].insert(0),cl[i].insert(n+1);
fo(i,1,n) cl[col[z[i]]].insert(i),modify(z[i],1);
int ans=0;
while(q--){
int op,u;
scanf("%d %d",&op,&u),u^=ans*t;
if(op==1){
int l,r;
scanf("%d %d",&l,&r),l^=ans*t,r^=ans*t;
printf("%d\n",ans=sum(r,u)-sum(l-1,u));
}
else{
int c;
scanf("%d",&c),c^=ans*t;
modify(u,-1),cl[col[u]].erase(fr[u]);
cl[col[u]=c].insert(fr[u]),modify(u,1);
}
}
}