题意

给出一个有根树,给出m个操作,操作1表示使得这一棵树的某一个节点的所有子树染上某一种颜色,操作2表示查询这个节点及其子树的颜色数量。

思路

对于一颗有根树来说,我们依靠dfs序就可以变换成为一个有序的序列,所以根据dfs序可以建立线段树。
对于线段树的操作,更新的话,区间更新,推lazy标记,问题是怎么去统计颜色的数量,这也关系到我们将怎么去推lazy标记。
如果单纯的修改,变换tree[root]=val这样,我们合并区间的时候,对于叶子节点的父亲来说可以合并,因为可以判断两个叶子是否相等,从而使得数量++,然而对于叶子的父亲的父亲,便没有办法统计不同的数目了。
我们进一步发现颜色只有60种,明显暗示你,状压思想。
所以改一下传统线段树更新方式,我们用位操作来更新即可,每一次取出来压好的状态,遍历1-60,有1的话答案加1即可。
ps:写这么久才发现的一个小tips:(wa了一发再tree更新上,注意线段树更新的时候,一般来讲如果区间更新的时候tree+=或者tree|=了,那么就是说直接对于这个区间做了影响,他的子树对他没有影响,所以没有pushup,同样,有pushup的时候,tree区间更新用=号。)

    #include<iostream>
    #include<cstring>
    #include<algorithm>
    #include<cmath>
    #include<cstdlib>
    #include<climits>
    #include<stack>
    #include<vector>
    #include<queue>
    #include<set>
    #include<map>
    //#include<regex>
    #include<cstdio>
    #define up(i,a,b)  for(int i=a;i<b;i++)
    #define dw(i,a,b)  for(int i=a;i>b;i--)
    #define upd(i,a,b) for(int i=a;i<=b;i++)
    #define dwd(i,a,b) for(int i=a;i>=b;i--)
    //#define local
    typedef long long ll;
    const double esp = 1e-6;
    const double pi = acos(-1.0);
    const int INF = 0x3f3f3f3f;
    const int inf = 1e9;
    using namespace std;
    int read()
    {
        char ch = getchar(); int x = 0, f = 1;
        while (ch<'0' || ch>'9') { if (ch == '-')f = -1; ch = getchar(); }
        while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0'; ch = getchar(); }
        return x * f;
    }
    typedef pair<int, int> pir;
    #define lson l,mid,root<<1
    #define rson mid+1,r,root<<1|1
    #define lrt root<<1
    #define rrt root<<1|1
    int n, m;
    const int N = 4e5 + 10;
    ll lazy[N<<2];
    ll tree[N<<2];
    int st[N], ed[N];
    struct {
        int to, next;
    }edge[2*N];
    int head[N];
    int cnt = 0;
    int a[N];
    void addedge(int u, int v)
    {
        edge[cnt].to = v;
        edge[cnt].next = head[u];
        head[u] = cnt++;
    }
    int num = 0;
    void dfs(int u,int f)
    {
        st[u] = ++num;
        for (int i = head[u]; ~i; i = edge[i].next)
        {
            if (edge[i].to != f)dfs(edge[i].to,u);
        }
        ed[u] = num;
    }
    void pushup(int root)
    {
        ll temp = 0; temp |= tree[lrt]; temp |= tree[rrt];
        tree[root] = temp;
    }
    void pushdown(int root)
    {
        if (lazy[root])
        {
            lazy[lrt] = lazy[root];
            lazy[rrt] = lazy[root];
            tree[lrt] = lazy[root];
            tree[rrt] = lazy[root];
            lazy[root] = 0;
        }
    }
    void build(int l, int r, int root)
    {
        lazy[root] = 0;
        if (l == r)
        {
            tree[root] = 0;
            return;
        }
        int mid = (l + r) >> 1;
        build(lson);
        build(rson);
        pushup(root);
    }
    void update(int l, int r, int root, int lf, int rt,int val)
    {
        if (lf <= l && r <= rt)
        {
            tree[root] = (1ll << val);
            lazy[root] = (1ll << val);
            return;
        }
        pushdown(root);
        int mid = (l + r) >> 1;
        if (lf <= mid)update(lson, lf, rt, val);
        if (rt > mid)update(rson, lf, rt, val);
        pushup(root);
    }
    ll querry(int l, int r, int root, int lf, int rt)
    {
        if (lf <= l && r <= rt)
        {
            return tree[root];
        }
        pushdown(root);
        int mid = (l + r) >> 1;
        ll ans = 0;
        if (lf <= mid)ans |= querry(lson, lf, rt);
        if (rt > mid)ans |= querry(rson, lf, rt);
        return ans;
    }
    int main()
    {
        n = read(); m = read();
        int x, y;
        memset(head, -1, sizeof(head));
        up(i, 0, n)a[i] = read();
        up(i, 0, n - 1)
        {
            x = read(), y = read();
            addedge(x, y); addedge(y, x);
        }
        dfs(1, 0);
        build(1, n, 1);
        up(i, 0, n)update(1, n, 1, st[i + 1], st[i + 1], a[i]);
        int op;
        while (m--)
        {
            op = read();
            if (op == 1)
            {
                x = read(), y = read();
                update(1, n, 1, st[x], ed[x], y);
            }
            else
            {
                x = read();
                ll ans = querry(1, n, 1, st[x], ed[x]);
                int temp = 0;
                upd(i, 1, 61)
                {
                    if ((1ll << i)&ans)temp++;
                }
                printf("%d\n", temp);
            }
        }
        return 0;
    }
02-10 22:39