CF620E New Year Tree 线段树+dfs序+bitset

线段树维护 dfs 序是显然的. 

暴力建 60 个线段树太慢，于是用 bitset 优化就好了 ~

code: 

#include <bits/stdc++.h>  
#define M 63      
#define N 800005    
#define lson now<<1 
#define rson now<<1|1    
using namespace std;         
inline void setIO(string s) 
{
    string in=s+".in"; 
    string out=s+".out"; 
    freopen(in.c_str(),"r",stdin);  
    freopen(out.c_str(),"w",stdout);   
}
namespace IO 
{ 
    char *p1,*p2,buf[100000];
    #define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
    int rd() {int x=0; char c=nc(); while(c<48) c=nc(); while(c>47) x=(((x<<2)+x)<<1)+(c^48),c=nc(); return x;}
}; 
int tot,edges; 
int dfn[N],size[N],st[N],ed[N],hd[N],to[N<<1],nex[N<<1],val[N];         
bitset<M>a[M],aa,a0;           
struct node 
{   
    bitset<M>v;      
    int tag; 
}t[N<<2];   
void add(int u,int v) 
{
    nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;   
}
void mark(int now,int v) 
{
    t[now].v=a[v];  
    t[now].tag=v;           
}
void pushdown(int l,int r,int now) 
{
    int mid=(l+r)>>1; 
    if(t[now].tag) 
    {
        mark(lson,t[now].tag);  
        if(r>mid) mark(rson,t[now].tag);   
        t[now].tag=0;                 
    }
}
void pushup(int l,int r,int now) 
{
    int mid=(l+r)>>1;  
    t[now].v=t[lson].v;  
    if(r>mid) t[now].v|=t[rson].v;  
}   
void update(int l,int r,int now,int L,int R,int v) 
{
    if(l>=L&&r<=R) 
    {
        mark(now,v);        
        return;   
    }
    pushdown(l,r,now);   
    int mid=(l+r)>>1;   
    if(L<=mid) update(l,mid,lson,L,R,v);  
    if(R>mid)  update(mid+1,r,rson,L,R,v);   
    pushup(l,r,now);   
}
void query(int l,int r,int now,int L,int R) 
{
    if(l>=L&&r<=R) 
    {
        aa|=t[now].v;         
        return;  
    }
    pushdown(l,r,now);   
    int mid=(l+r)>>1;   
    if(L<=mid) query(l,mid,lson,L,R); 
    if(R>mid)  query(mid+1,r,rson,L,R);     
}
void dfs(int u,int ff) 
{ 
    size[u]=1;   
    st[u]=dfn[u]=++tot; 
    for(int i=hd[u];i;i=nex[i]) 
    { 
        int v=to[i]; 
        if(v==ff) continue;       
        dfs(v,u);  
        size[u]+=size[v];  
    } 
    ed[u]=tot;    
}
int main() 
{
    // setIO("data-structure"); 
    int i,j,n,m;    
    n=IO::rd();
    m=IO::rd();    
    // scanf("%d%d",&n,&m);        
    for(i=1;i<=61;++i) a[i][i]=1;                
    for(i=1;i<=n;++i) val[i]=IO::rd();    
    for(i=1;i<n;++i) 
    {
        int u=IO::rd(),v=IO::rd();         
        // scanf("%d%d",&u,&v); 
        add(u,v),add(v,u);   
    }    
    dfs(1,0);              
    for(i=1;i<=n;++i) update(1,n,1,st[i],st[i],val[i]);    
    for(i=1;i<=m;++i) 
    {
        int opt=IO::rd(),x=IO::rd(),y;        
        if(opt==1) 
        {
            y=IO::rd();    
            update(1,n,1,st[x],ed[x],y);  
        }
        else 
        {
            aa=a0; 
            query(1,n,1,st[x],ed[x]);         
            printf("%d
",(int)aa.count());     
        }
    }
    return 0; 
}