Z - New Year Tree CodeForces - 620E 线段树 区间种类 bitset

Z - New Year Tree

CodeForces - 620E

这个题目还没有写，先想想思路，我觉得这个题目应该可以用bitset，

首先这个肯定是用dfs序把这个树转化成线段树，也就是二叉树。

然后就是一个区间修改和区间查询。这个区间查询时查询这个区间的种类数。

这个之前写过几个题目也是查区间种类数的

G. Yash And Trees 线段树 bitset

20190709 暑训 区间种类数 莫队的学习

莫队和权值线段树应该都是不支持修改的，所以这个题目用不上，

然后就是这个高端压位卡常容器bitset 我觉得这个题目应该可以用到。

#include <cstring>
#include <queue>
#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <string>
#include <bitset>
#include <algorithm>
#include <map>
#include <vector>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
typedef long long ll;
typedef bitset<> bit;
const int maxn = 4e5 + ;
vector<int>G[maxn];
ll el[maxn], er[maxn], a[maxn], tot, num[maxn];

void dfs(int u,int pre)
{
    el[u] = ++tot;
    num[tot] = u;
    for(int i=;i<G[u].size();i++)
    {
        int v = G[u][i];
        if (v == pre) continue;
        dfs(v, u);
    }
    er[u] = tot;
}

bit tree[maxn * ];
ll lazy[maxn * ];

void build(int id,int l,int r)
{
    lazy[id] = ;
    if(l==r)
    {
        tree[id].reset();
        tree[id] = (1ll << (a[num[l]] - ));
        return;
    }
    int mid=(l + r) >> ;
    build(id << , l, mid);
    build(id <<  | , mid + , r);
    tree[id] = tree[id << ] | tree[id <<  | ];
}

void push_down(int id)
{
    if(lazy[id]!=)
    {
        tree[id << ].reset(); tree[id <<  | ].reset();
        tree[id << ] = (1ll << (lazy[id] - ));
        tree[id <<  | ] = (1ll << (lazy[id] - ));
        lazy[id << ] = lazy[id <<  | ] = lazy[id];
        lazy[id] = ;
    }
}

void update(int id,int l,int r,int x,int y,int val)
{
    if(x<=l&&y>=r)
    {
        tree[id].reset();
        tree[id] = (1ll << (val - ));
        lazy[id] = val;
        return;
    }
    push_down(id);
    int mid = (l + r) >> ;
    if (x <= mid) update(id << , l, mid, x, y, val);
    if (y > mid) update(id <<  | , mid + , r, x, y, val);
    tree[id] = tree[id << ] | tree[id <<  | ];
}

bit query(int id,int l,int r,int x,int y)
{
    if (x <= l && y >= r) return tree[id];
    int mid = (l + r) >> ;
    bit ans;
    ans.reset();
    push_down(id);
    if (x <= mid) ans = query(id << , l, mid, x, y);
    if (y > mid) ans |= query(id <<  | , mid + , r, x, y);
    return ans;
}

int main() {
    int n, m;
    tot = ;
    scanf("%d%d", &n, &m);
    for (int i = ; i <= n; i++) scanf("%lld", &a[i]);
    for (int i = ; i < n; i++) {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(, -);
    build(, , n);
    while (m--) {
        int opt, v, x;
        scanf("%d", &opt);
        if (opt == ) {
            scanf("%d%d", &v, &x);
            update(, , n, el[v], er[v], x);
        }
        else {
            scanf("%d", &v);
            bit ans = query(, , n, el[v], er[v]);
            printf("%d\n", ans.count());
        }
    }
    return ;
}