CF--- PROBLEM 620 E 【思维 + DFS序 + 二进制状态】

传送门
//题意: 对于一棵树以1为根的树, 每个结点有一种颜色. 现在有两种操作.
1 x y 把以x为根节点的子树中的所有节点染成颜色y
2 x 询问以x为根节点的子树中有多少种颜色的节点
//思路: 很明显的DFS序, 需要考虑几点, 涉及到线段树的区间修改. 然后就是颜色的计数, 由于颜色最多60种, 那么用long long 的二进制来表示它的颜色. 这样再pushup时直接左右子树或起来就行. 注意一个坑点就是在DFS序中的线段树描述的是各个节点的序了!!! 不再是每一个节点本身. 所以在build中, tree[id].val = a[dfsxu[l]] (dfsxu表示的是当前节点在dfs序中代表的节点!!!), 而不是a[l] !!! 这个BUG调了我好久. 要多做做这方面的题啊.

AC Code

/** @Cain*/const int maxn = 4e5+5;int cas=1;ll a[maxn];int dfs_xx[maxn];struct node{    int tl, tr; ll val, maxx, lazy;    void fun(ll tmp) {      //如果是直接加值就用 +=        lazy = tmp;        val = 1ll<<(tmp-1);        //maxx = tmp;    }} tree[maxn*4];void pushup(int id){    tree[id].val = tree[id<<1].val|tree[id<<1|1].val;    //tree[id].maxx = max(tree[id<<1].maxx,tree[id<<1|1].maxx);}void pushdown(int id){    if(tree[id].lazy){        tree[id<<1].fun(tree[id].lazy);        tree[id<<1|1].fun(tree[id].lazy);        tree[id].lazy = 0;    }}void build(int id,int l,int r){    tree[id].tl = l; tree[id].tr = r; tree[id].lazy = 0;    if(l == r){        tree[id].val = a[dfs_xx[l]];  //这是一个坑点啊.就如上面我所说.        return ;    }    int mid = (l+r) >> 1;    build(id<<1,l,mid);    build(id<<1|1,mid+1,r);    pushup(id);}void update(int id,int st,int ed,int val){    int l=tree[id].tl, r=tree[id].tr;    if(st <= l && r <= ed){        tree[id].fun(val);        return ;    }    pushdown(id);    int mid = (l+r) >> 1;    if(st <= mid) update(id<<1,st,ed,val);    if(ed > mid) update(id<<1|1,st,ed,val);    pushup(id);}ll query(int id,int ql,int qr){    int l = tree[id].tl , r = tree[id].tr;    if(ql <= l && r <= qr) return tree[id].val;    pushdown(id);    int mid = (l+r) >> 1 ;    ll res = 0;    if(ql <= mid) res |= query(id<<1,ql,qr);    if(qr > mid) res |= query(id<<1|1,ql,qr);    return res;}struct edge{    int to,next;    ll w;}s[maxn];int cnt,ti,head[maxn];int p1[maxn],p2[maxn];void add(int u,int v,ll w){    s[cnt] = (edge){v,head[u],w};    head[u] = cnt++;}void dfs_id(int u,int fa){    p1[u] = ++ti;    dfs_xx[ti] = u;    for(int i=head[u]; ~i ; i = s[i].next){        int to = s[i].to;        if(fa == to) continue;        dfs_id(to,u);    }    p2[u] = ti;}void solve(){    int n,q;    ll b[50];    scanf("%d%d",&n,&q);    for(int i=1;i<=n;i++){        ll u; scanf("%lld",&u);        b[i] = u;        a[i] = 1ll<<(u-1);    }    ti = cnt = 0; Fill(head,-1);    for(int i=1;i<=n-1;i++){        int u,v;        scanf("%d%d",&u,&v);        add(u,v,1);        add(v,u,1);    }    dfs_id(1,-1);    build(1,1,n);    while(q--){        int op; scanf("%d",&op);        if(op == 1) {            int u,v;            scanf("%d%d",&u,&v);            update(1,p1[u],p2[u],v);        }        else {            int u; scanf("%d",&u);            ll x = query(1,p1[u],p2[u]);            int cnt = 0;            while(x){                if(x&1) cnt++;                x >>= 1;            }            printf("%d\n",cnt);        }    }}