题解
弱智题，二进制表示位数。合并时用｜
就是被1<<x卡了好久。
要写成1ll<<x才行
1 #include<iostream> 2 #include<cstring> 3 #include<cmath> 4 #include<cstdio> 5 #include<algorithm> 6 using namespace std; 7 const int N=400010; 8 int cnt,head[N]; 9 int id[N],size[N],cao[N],tot; 10 int n,m,a[N]; 11 struct edge{ 12 int to,nxt; 13 }e[N*2]; 14 struct tree{ 15 int l,r,lazy; 16 long long sum; 17 }tr[N*9]; 18 void add(int u,int v){ 19 cnt++; 20 e[cnt].nxt=head[u]; 21 e[cnt].to=v; 22 head[u]=cnt; 23 } 24 void dfs1(int u,int fa){ 25 id[u]=++tot; 26 cao[tot]=u; 27 size[u]=1; 28 for(int i=head[u];i;i=e[i].nxt){ 29 int v=e[i].to; 30 if(v==fa)continue; 31 dfs1(v,u); 32 size[u]+=size[v]; 33 } 34 } 35 void build(int l,int r,int now){ 36 tr[now].l=l; 37 tr[now].r=r; 38 if(l==r){ 39 tr[now].sum=1ll<<a[cao[l]]; 40 return; 41 } 42 int mid=(l+r)>>1; 43 build(l,mid,now*2); 44 build(mid+1,r,now*2+1); 45 tr[now].sum=tr[now*2].sum|tr[now*2+1].sum; 46 } 47 void pushdown(int now){ 48 if(tr[now].lazy==0)return; 49 tr[now*2].sum=tr[now*2+1].sum=1ll<<tr[now].lazy; 50 tr[now*2].lazy=tr[now*2+1].lazy=tr[now].lazy; 51 tr[now].lazy=0; 52 } 53 void update(int l,int r,int now,int c){ 54 pushdown(now); 55 if(tr[now].l==l&&tr[now].r==r){ 56 tr[now].sum=1ll<<c; 57 tr[now].lazy=c; 58 return; 59 } 60 int mid=(tr[now].l+tr[now].r)>>1; 61 if(l>mid)update(l,r,now*2+1,c); 62 else if(r<=mid)update(l,r,now*2,c); 63 else{ 64 update(l,mid,now*2,c); 65 update(mid+1,r,now*2+1,c); 66 } 67 tr[now].sum=tr[now*2].sum|tr[now*2+1].sum; 68 } 69 long long query(int l,int r,int now){ 70 pushdown(now); 71 if(tr[now].l==l&&tr[now].r==r){ 72 return tr[now].sum; 73 } 74 int mid=(tr[now].l+tr[now].r)>>1; 75 if(l>mid)return query(l,r,now*2+1); 76 else if(r<=mid)return query(l,r,now*2); 77 else{ 78 return query(l,mid,now*2)|query(mid+1,r,now*2+1); 79 } 80 } 81 int work(long long x){ 82 int ans=0; 83 while(x){ 84 if(x&1)ans++; 85 x>>=1; 86 } 87 return ans; 88 } 89 int main(){ 90 scanf("%d%d",&n,&m); 91 for(int i=1;i<=n;i++){ 92 scanf("%d",&a[i]); 93 } 94 for(int i=1;i<n;i++){ 95 int u,v; 96 scanf("%d%d",&u,&v); 97 add(u,v);add(v,u); 98 } 99 dfs1(1,0); 100 build(1,n,1); 101 for(int i=1;i<=m;i++){ 102 int k; 103 scanf("%d",&k); 104 if(k==1){ 105 int u,c; 106 scanf("%d%d",&u,&c); 107 // cout<<id[u]<<" "<<id[u]+size[u]-1<<endl; 108 update(id[u],id[u]+size[u]-1,1,c); 109 } 110 else{ 111 int u; 112 scanf("%d",&u); 113 printf("%d ",work(query(id[u],id[u]+size[u]-1,1))); 114 } 115 } 116 return 0; 117 }