做法:
因为只有60种颜色,所以可以状压颜色,这样的话就可以把区间颜色合并的操作看为“或”,线段树随便维护一下就可以了。
WA了无数发才想起来 ”1LL << 一个东西“ 和 “1 << 一个long long 的东西” 是不一样的(打多校的时候还WA过一发)……贼气.
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=400005;
struct edge{
int to,nxt;
}e[maxn<<1];
struct node{
int l,r;
unsigned long long num,tag;
}tr[maxn<<2];
int L[maxn],R[maxn],rnk[maxn];
int n,m,ecnt;
int head[maxn];
unsigned long long tmp[maxn];
void ins(int u,int v){
e[ecnt].to=v;
e[ecnt].nxt=head[u];
head[u]=ecnt++;
}
int cnt;
void dfs(int u,int fa){
L[u]=++cnt;
rnk[cnt]=u;
for(int i=head[u];~i;i=e[i].nxt){
if(e[i].to==fa) continue;
dfs(e[i].to,u);
}
R[u]=cnt;
}
void push_up(int k){tr[k].num=tr[k<<1].num|tr[k<<1|1].num;}
void push_down(int k){
if(tr[k].tag){
tr[k<<1].num=tr[k<<1|1].num=tr[k].tag;
tr[k<<1].tag=tr[k<<1|1].tag=tr[k].tag;
tr[k].tag=0;
}
}
void build(int k,int l,int r){
tr[k].l=l,tr[k].r=r;
if(l==r){
tr[k].num=1LL<<tmp[rnk[l]];
tr[k].tag=0;
return;
}
int mid=(l+r)>>1;
build(k<<1,l,mid),build(k<<1|1,mid+1,r);
push_up(k);
}
void update(int k,int L,int R,unsigned long long x){
int l=tr[k].l,r=tr[k].r;
if(l==r){tr[k].num=x;return;}
if(L<=l&&R>=r){tr[k].tag=tr[k].num=x;return;}
int mid=(l+r)>>1;
push_down(k);
if(R<=mid) update(k<<1,L,R,x);
else if(L>mid) update(k<<1|1,L,R,x);
else update(k<<1,L,R,x),update(k<<1|1,L,R,x);
push_up(k);
}
unsigned long long query(int k,int L,int R){
int l=tr[k].l,r=tr[k].r;
if(L<=l&&R>=r) return tr[k].num;
int mid=(l+r)>>1;
push_down(k);
if(R<=mid) return query(k<<1,L,R);
else if(L>mid) return query(k<<1|1,L,R);
else return query(k<<1,L,R)|query(k<<1|1,L,R);
push_up(k);
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {head[i]=-1;scanf("%I64u",&tmp[i]);}
int uu,vv;
for(int i=1;i<n;i++){
scanf("%d%d",&uu,&vv);
ins(uu,vv),ins(vv,uu);
}
dfs(1,-1);
build(1,1,n);
int op,a;
unsigned long long b;
for(int i=1;i<=m;i++){
scanf("%d",&op);
if(op==1){
scanf("%d%I64u",&a,&b);
update(1,L[a],R[a],(1LL<<b));
}else{
scanf("%d",&a);
printf("%d\n",__builtin_popcountll(query(1,L[a],R[a])));
}
}
return 0;
}