Codeforces 620E New Year Tree

http://codeforces.com/problemset/problem/620/E

题意：
给以一棵树，每个结点刚开始的时候都有一个颜色，现在有查询1 u col：给这个结点及其子树染上col这种颜色，2 u：查询以u为根节点的子树的所有颜色种类

分析：
显然是到线段树的题目，要把每个节点的子树放到一个区间上，并且记录首尾位置，这样就可以更新颜色的时候成段更新，查询颜色种类的时候区间查询。所以用dfs遍历整棵树，fir数组记录根节点位置，la数组保存子树的最后一个位置，用个数组当做哈希索引ha，保存线段树每个点对应的树上的某个节点，然后就是基本的线段树操作了。
PS：
对于颜色的保存，因为只有60种颜色，用二进制就好了，如果更多的话可以用字符串，也可以用数组。
代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N=4e5+5;
vector<int>e[N];
int fir[N],la[N],a[N],ha[N];
bool vis[N];
int id;
void dfs(int u)
{
    fir[u]=id; ha[id]=u; vis[u]=1;
    for(int i=0;i<e[u].size();i++){
        if(!vis[e[u][i]]){
            ++id;dfs(e[u][i]);
        }
    }
    la[u]=id;
}
ll sum[N<<2]; //保存所有节点的颜色
int col[N<<2]; //标记子树涂上什么颜色
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}
void pushdown(int rt)
{
    if(col[rt]){
        col[rt<<1]=col[rt<<1|1]=col[rt];
        sum[rt<<1]=sum[rt<<1|1]=1ll<<col[rt];
        col[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    if(l==r){
        sum[rt]=1ll<<(a[ha[l]]);
        col[rt]=0;return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int a,int b,int c,int l,int r,int rt)
{
    if(a<=l&&r<=b){
        sum[rt]=1ll<<c;
        col[rt]=c;
        return;
    }
    int m=(l+r)>>1;
    pushdown(rt);
    if(a<=m)update(a,b,c,lson);
    if(b>m)update(a,b,c,rson);
    pushup(rt);
}
ll query(int a,int b,int l,int r,int rt)
{
    if(a<=l&&r<=b){
        return sum[rt];
    }
    int m=(l+r)>>1;
    pushdown(rt);
    ll ans=0;
    if(a<=m)ans|=query(a,b,lson);
    if(b>m)ans|=query(a,b,rson);
    return ans;
}
int main()
{
    int n,m,u,v;
    //freopen("f.txt","r",stdin);
    scanf("%d%d",&n,&m);
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        e[u].push_back(v);e[v].push_back(u);
    }
    id=1;
    dfs(1);
    build(1,n,1);
    while(m--){
        int op,col;
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d",&u,&col);
            update(fir[u],la[u],col,1,n,1);
        }
        else{
            scanf("%d",&u);
            ll ans=query(fir[u],la[u],1,n,1);
            int tot=0;
            while(ans){
                if(ans&1)tot++;
                ans>>=1;
            }
            printf("%d\n",tot);
        }
    }
    return 0;
}