Educational Codeforces Round 6 E. New Year Tree（DFS序+线段树）

题意：给你一棵树，编号1~n，告诉你根结点是1。 每次有两个操作：

1，将以v为根的子树的结点全部染成颜色c

2，问以v为根的紫书的结点的颜色种类。

思路：如果这是一条线段的话， 那么这就是线段树的区间更新问题，而现在是一棵树。

因为告诉了根结点是1， 那么这棵树的任意一个结点的子树就是确定的， 所以我们可以用DFS的先序遍历，将所有结点重新编号，因为先序遍历的话， 任意一个结点和其子树的编号就是一条连续的线段了，在这其中维护每个结点的新编号， 和这个结点的子树中的最大编号即可。

然后就是线段树区间更新了， 由于颜色数最大60， 用long long通过位运算的 | 操作就行了， 注意对1左移的时候应该先将1转成long long再进行操作。

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF  0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-9
#define maxn 1000100
#define MOD 1000000007


struct Edge
{
    int to,next;
}edge[maxn];
int n,m,tot,pos,num[maxn],mp[maxn];
int head[maxn],tmp[maxn],a[maxn];
void init()
{
    tot = 0;
    memset(head,-1,sizeof(head));
}
void add_edge(int u,int v)
{
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}
void dfs(int u,int pre)
{
    num[u] = 1;
    tmp[u] = pos;
    mp[pos++] = u;
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre)
            continue;
        dfs(v,u);
        num[u] += num[v];
    }
}

struct node
{
    int l,r;
    long long sum,lazy;
}tree[maxn*4];

void pushup(int pos)
{
    tree[pos].sum = tree[pos<<1].sum | tree[pos<<1|1].sum;
}
void pushdown(int pos)
{
    if(tree[pos].lazy == 0)
        return;
    tree[pos<<1].sum = tree[pos].sum;
    tree[pos<<1|1].sum = tree[pos].sum;
    tree[pos<<1].lazy = 1;
    tree[pos<<1|1].lazy = 1;
    tree[pos].lazy = 0;
}

void build(int l,int r,int pos)
{
    tree[pos].l = l;
    tree[pos].r = r;
    tree[pos].sum = 0;
    tree[pos].lazy = 0;
    if(l == r)
    {
        tree[pos].sum = 1ll<<(a[mp[l]]-1);
        return;
    }
    int mid = (l+r)>>1;
    build(l,mid,pos<<1);
    build(mid+1,r,pos<<1|1);
    pushup(pos);
}                                                   //建树                                                  //查询操作

void update(int l,int r,int pos,long long add)
{
    if(tree[pos].l == l && tree[pos].r == r)
    {
        tree[pos].sum = 1ll<<(add-1);
        tree[pos].lazy = 1;
        return ;
    }
    pushdown(pos);
    int mid = (tree[pos].l + tree[pos].r) >> 1;
    if(r <= mid)
        update(l,r,pos<<1,add);
    else if(l > mid)
        update(l,r,pos<<1|1,add);
    else
    {
        update(l,mid,pos<<1,add);
        update(mid+1,r,pos<<1|1,add);
    }
    pushup(pos);
}                                         //自上而下更新节点

long long query(int l,int r,int pos)
{
    if(l == tree[pos].l && r == tree[pos].r)
        return tree[pos].sum;
    pushdown(pos);
    int mid = (tree[pos].l+tree[pos].r)>>1;
    if(r <= mid)
        return query(l,r,pos<<1);
    else if(l > mid)
        return query(l,r,pos<<1|1);
    else
        return query(l,mid,pos<<1) | query(mid+1,r,pos<<1|1);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t,C = 1;
    //scanf("%d",&t);
    while(scanf("%d%d",&n,&m) != EOF)
    {
        init();
        pos = 1;
        memset(num,0,sizeof(num));
        for(int i = 1; i <= n; i++)
            scanf("%d",&a[i]);
        for(int i = 0; i < n-1; i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            add_edge(x,y);
            add_edge(y,x);
        }
        dfs(1,-1);
        build(1,n,1);
        for(int i = 0; i < m; i++)
        {
            int x;
            scanf("%d",&x);
            if(x == 1)
            {
                int k,c;
                scanf("%d%d",&k,&c);
                update(tmp[k],tmp[k]+num[k]-1,1,c);
            }
            else
            {
                int k;
                scanf("%d",&k);
                long long cnt = query(tmp[k],tmp[k]+num[k]-1,1);
                int ans = 0;
                while(cnt)
                {
                    if(cnt & 1)
                        ans++;
                    cnt >>= 1;
                }
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}