CodeForces 620E New Year Tree

线段树+位运算

首先对树进行DFS，写出DFS序列，记录下每一个节点控制的区间范围。然后就是区间更新和区间查询了。

某段区间的颜色种类可以用位运算来表示，方便计算。

如果仅有第i种颜色，那么就用十进制数（1<<i）表示。

如果A区间有的颜色是col1，B区间有的颜色是col2，合并之后有的就是（col1 | col2）

输出有几种，就是看得到的十进制数的二进制表示中有多少位是1.

#include<cstdio>
#include<cstring>
#include<cmath>
#include<stack>
#include<vector>
#include<map>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn=*+;
int n,q;
int col[maxn];
vector<int> Tree[maxn];
bool b[maxn];
int Left[maxn],Right[maxn];
int s[*maxn];
int time;
long long ans;
struct SegTree
{
    bool flag;
    long long ans;
}segTree[*maxn*];

void dfs(int now)
{
    b[now]=;
    Left[now]=(++time);
    s[time]=now;
    for(int i=;i<Tree[now].size();i++)
        if(b[Tree[now][i]]==)
            dfs(Tree[now][i]);
    Right[now]=(++time);
    s[time]=now;
}

void pushUp(int rt)
{
    segTree[rt].ans=(segTree[*rt].ans|segTree[*rt+].ans);
}

void pushDown(int rt)
{
    if(segTree[rt].flag!=)
    {
        segTree[*rt].flag=segTree[*rt+].flag=segTree[rt].flag;
        segTree[*rt].ans=segTree[*rt+].ans=segTree[rt].ans;
        segTree[rt].flag=;
    }
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        segTree[rt].flag=;
        segTree[rt].ans=(long long)<<((long long)col[s[l]]);
        return ;
    }
    int m=(l+r)/;
    if(l<=m) build(l,m,*rt);
    if(r>m) build(m+,r,*rt+);
    pushUp(rt);
    return;
}

void quary(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        ans=(ans|segTree[rt].ans);
        return;
    }
    pushDown(rt);
    int m=(l+r)/;
    if(L<=m) quary(L,R,l,m,*rt);
    if(R>m) quary(L,R,m+,r,*rt+);
    pushUp(rt);
    return;
}

void update(int info,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        segTree[rt].flag=info;
        segTree[rt].ans=(long long)<<(long long) info;
        return;
    }

    pushDown(rt);
    int m=(l+r)/;
    if(L<=m) update(info,L,R,l,m,*rt);
    if(R>m) update(info,L,R,m+,r,*rt+);
    pushUp(rt);
}

int main()
{
    scanf("%d%d",&n,&q);
    for(int i=;i<=n;i++) scanf("%d",&col[i]);
    for(int i=;i<=n;i++) Tree[i].clear();
    for(int i=;i<=n-;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        Tree[x].push_back(y);
        Tree[y].push_back(x);
    }
    memset(b,,sizeof b);
    time=,dfs();
    build(,*n,);
    for(int i=;i<=q;i++)
    {
        int tk;
        scanf("%d",&tk);
        if(tk==)
        {
            int vk,ck;
            scanf("%d%d",&vk,&ck);
            update(ck,Left[vk],Right[vk],,*n,);
        }
        else
        {
            int vk;
            scanf("%d",&vk);
            ans=;
            quary(Left[vk],Right[vk],,*n,);
            int num=;
            while(ans)
            {
                if(ans%==) num++;
                ans=ans/;
            }
            printf("%d\n",num);
        }
    }
    return ;
}