一本通1760:树上数颜色
链接
将相同颜色的点按 \(dfs\) 序排序,在树上差分。
因为有颜色和 \(dfs\) 序两个维度,用树状数组套动态开点线段树维护。
#pragma G++ optimize(2)
#include<bits/stdc++.h>
#define IL inline
#define LL long long
using namespace std;
const int N=1e5+3;
struct hh{int to,nxt;}e[N<<1];
int n,m,fir[N],num,col[N],dfn[N],rev[N],dep[N],son[N],siz[N],fa[N],top[N];
int cnt,ls[N*400],rs[N*400],sum[N*400],rt[N],ans;
set<int>st[N];
set<int>::iterator it1,it2,it3,it4;
IL int in(){
char c;int f=1;
while((c=getchar())<'0'||c>'9')
if(c=='-') f=-1;
int x=c-'0';
while((c=getchar())>='0'&&c<='9')
x=x*10+c-'0';
return x*f;
}
IL void link(int x,int y){e[++num]=(hh){y,fir[x]},fir[x]=num;}
void dfs1(int u,int f){
fa[u]=f,dep[u]=dep[f]+1,siz[u]=1;
for(int i=fir[u],v;v=e[i].to;i=e[i].nxt)
if(v^f){
dfs1(v,u),siz[u]+=siz[v];
if(siz[son[u]]<siz[v]) son[u]=v;
}
}
void dfs2(int u,int t){
top[u]=t,rev[dfn[u]=++num]=u;
if(son[u]) dfs2(son[u],t);
for(int i=fir[u],v;v=e[i].to;i=e[i].nxt)
if(v^fa[u]&&v^son[u]) dfs2(v,v);
}
int get_lca(int x,int y){
while(top[x]^top[y]) dep[top[x]]<dep[top[y]]?y=fa[top[y]]:x=fa[top[x]];
return dep[x]<dep[y]?x:y;
}
void mdy(int &o,int l,int r,int u,int op){
if(!o) o=++cnt;sum[o]+=op;
if(l==r) return;
int mid=l+r>>1;
if(u<=mid) mdy(ls[o],l,mid,u,op);
else mdy(rs[o],mid+1,r,u,op);
}
int query(int o,int l,int r,int ll,int rr){
if(!o) return 0;
if(l>=ll&&r<=rr) return sum[o];
int mid=l+r>>1;
if(rr<=mid) return query(ls[o],l,mid,ll,rr);
if(ll>mid) return query(rs[o],mid+1,r,ll,rr);
return query(ls[o],l,mid,ll,rr)+query(rs[o],mid+1,r,ll,rr);
}
IL int lb(int x){return x&-x;}
IL void add(int y,int x,int op){for(;y<=n;y+=lb(y)) mdy(rt[y],1,n,x,op);}
IL int ask(int y,int l,int r){int res=0;for(;y;y-=lb(y) ) res+=query(rt[y],1,n,l,r);return res;}
void Add(int x,int c){
it1=st[c].insert(dfn[x]).first,it2=st[c].end(),--it2,add(c,dfn[x],1);
if(it1!=st[c].begin()) it3=it1,--it3,add(c,dfn[get_lca(rev[*it1],rev[*it3])],-1);
if(it1!=it2) it4=it1,++it4,add(c,dfn[get_lca(rev[*it1],rev[*it4])],-1);
if(it1!=st[c].begin()&&it1!=it2) add(c,dfn[get_lca(rev[*it3],rev[*it4])],1);
}
void Del(int x,int c){
it1=st[c].find(dfn[x]),it2=st[c].end(),--it2,add(c,dfn[x],-1);
if(it1!=st[c].begin()) it3=it1,--it3,add(c,dfn[get_lca(rev[*it1],rev[*it3])],1);
if(it1!=it2) it4=it1,++it4,add(c,dfn[get_lca(rev[*it1],rev[*it4])],1);
if(it1!=st[c].begin()&&it1!=it2) add(c,dfn[get_lca(rev[*it3],rev[*it4])],-1);
st[c].erase(dfn[x]);
}
int Ask(int u,int l,int r){return ask(r,dfn[u],dfn[u]+siz[u]-1)-ask(l-1,dfn[u],dfn[u]+siz[u]-1);}
int main()
{
int op,t,x,y,c;
n=in(),m=in(),op=in();
for(int i=1;i<=n;++i) col[i]=in();
for(int i=1;i<n;++i)
x=in(),y=in(),
link(x,y),link(y,x);
num=0,dfs1(1,0),dfs2(1,1);
for(int i=1;i<=n;++i) Add(i,col[i]);
while(m--){
t=in(),x=in(),y=in();
if(op) x^=ans,y^=ans;
if(t==1) c=op?in()^ans:in(),printf("%d\n",ans=Ask(x,y,c));
else Del(x,col[x]),col[x]=y,Add(x,col[x]);
}
return 0;
}