New Year Tree CodeForces -620E

这个题有一个技巧：把颜色压到一个long long 上。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
typedef long long LL;
LL lft[],c[];
vector<int> e[];
int n,m;
int pp[],lp[],rp[],len;
void dfs(int u,int fa)
{
    pp[++len]=u;lp[u]=len;
    for(auto& v:e[u])
        if(v!=fa)
            dfs(v,u);
    rp[u]=len;
}
#define mid ((l+r)>>1)
#define lc (num<<1)
#define rc (num<<1|1)
LL dat[],setv[];
void build(int l,int r,int num)
{
    setv[num]=-;
    if(l==r)    {dat[num]=c[pp[l]];return;}
    build(l,mid,lc);build(mid+,r,rc);
    dat[num]=dat[lc]|dat[rc];
}
int L,R;LL x;
void pd(int l,int r,int num)
{
    if(setv[num]!=-)
    {
        dat[lc]=setv[num];dat[rc]=setv[num];
        setv[lc]=setv[num];setv[rc]=setv[num];
        setv[num]=-;
    }
}
void change(int l,int r,int num)
{
    if(L<=l&&r<=R)    {dat[num]=x;setv[num]=x;return;}
    pd(l,r,num);
    if(L<=mid)    change(l,mid,lc);
    if(mid<R)    change(mid+,r,rc);
    dat[num]=dat[lc]|dat[rc];
}
LL query(int l,int r,int num)
{
    if(L<=l&&r<=R)    return dat[num];
    pd(l,r,num);
    LL ans=;
    if(L<=mid)    ans|=query(l,mid,lc);
    if(mid<R)    ans|=query(mid+,r,rc);
    return ans;
}
int main()
{
    int i,a,b,v,idx;
    lft[]=;
    for(i=;i<=;i++)    lft[i]=lft[i-]<<;
    scanf("%d%d",&n,&m);
    for(i=;i<=n;i++)    scanf("%lld",&c[i]),c[i]=lft[c[i]];
    for(i=;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        e[a].push_back(b);e[b].push_back(a);
    }
    dfs(,);
    build(,n,);
    while(m--)
    {
        scanf("%d",&idx);
        if(idx==)
        {
            scanf("%d%lld",&v,&x);x=lft[x];
            L=lp[v];R=rp[v];change(,n,);
        }
        else if(idx==)
        {
            scanf("%d",&v);L=lp[v];R=rp[v];
            printf("%d\n",__builtin_popcountll(query(,n,)));
        }
    }
    return ;
}