CF620E New Year Tree(线段树+二进制）

题解

弱智题，二进制表示位数。合并时用｜

就是被1<<x卡了好久。

要写成1ll<<x才行

 #include<iostream>
 #include<cstring>
 #include<cmath>
 #include<cstdio>
 #include<algorithm>
 using namespace std;
 const int N=;
 int cnt,head[N];
 int id[N],size[N],cao[N],tot;
 int n,m,a[N];
 struct edge{
     int to,nxt;
 }e[N*];
 struct tree{
     int l,r,lazy;
     long long sum;
 }tr[N*];
 void add(int u,int v){
     cnt++;
     e[cnt].nxt=head[u];
     e[cnt].to=v;
     head[u]=cnt;
 }
 void dfs1(int u,int fa){
     id[u]=++tot;
     cao[tot]=u;
     size[u]=;
     for(int i=head[u];i;i=e[i].nxt){
         int v=e[i].to;
         if(v==fa)continue;
         dfs1(v,u);
         size[u]+=size[v];
     }
 }
 void build(int l,int r,int now){
     tr[now].l=l;
     tr[now].r=r;
     if(l==r){
         tr[now].sum=1ll<<a[cao[l]];
         return;
     }
     int mid=(l+r)>>;
     build(l,mid,now*);
     build(mid+,r,now*+);
     tr[now].sum=tr[now*].sum|tr[now*+].sum;
 }
 void pushdown(int now){
     if(tr[now].lazy==)return;
     tr[now*].sum=tr[now*+].sum=1ll<<tr[now].lazy;
     tr[now*].lazy=tr[now*+].lazy=tr[now].lazy;
     tr[now].lazy=;
 }
 void update(int l,int r,int now,int c){
     pushdown(now);
     if(tr[now].l==l&&tr[now].r==r){
         tr[now].sum=1ll<<c;
         tr[now].lazy=c;
         return;
     }
     int mid=(tr[now].l+tr[now].r)>>;
     if(l>mid)update(l,r,now*+,c);
     else if(r<=mid)update(l,r,now*,c);
     else{
         update(l,mid,now*,c);
         update(mid+,r,now*+,c);
     }
     tr[now].sum=tr[now*].sum|tr[now*+].sum;
 }
 long long query(int l,int r,int now){
     pushdown(now);
     if(tr[now].l==l&&tr[now].r==r){
         return tr[now].sum;
     }
     int mid=(tr[now].l+tr[now].r)>>;
     if(l>mid)return query(l,r,now*+);
     else if(r<=mid)return query(l,r,now*);
     else{
         return query(l,mid,now*)|query(mid+,r,now*+);
     }
 }
 int work(long long x){
     int ans=;
     while(x){
         if(x&)ans++;
         x>>=;
     }
     return ans;
 }
 int main(){
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++){
         scanf("%d",&a[i]);
     }
     for(int i=;i<n;i++){
         int u,v;
         scanf("%d%d",&u,&v);
         add(u,v);add(v,u);
     }
     dfs1(,);
     build(,n,);
     for(int i=;i<=m;i++){
         int k;
         scanf("%d",&k);
         if(k==){
             int u,c;
             scanf("%d%d",&u,&c);
         //    cout<<id[u]<<" "<<id[u]+size[u]-1<<endl;
             update(id[u],id[u]+size[u]-,,c);
         }
         else{
             int u;
             scanf("%d",&u);
             printf("%d\n",work(query(id[u],id[u]+size[u]-,)));
         }
     }
     return ;
 }