python/TangDou/AcWing/SegmentTree/P2787_XDS.cpp

#include <bits/stdc++.h>

using namespace std;
const int N = 1e5 + 10;

int n, m, f[26];
char a[N]; // 初始化输入的字符串

// 多棵线段树法
#define ls (u << 1)
#define rs (u << 1 | 1)
#define mid ((l + r) >> 1)
struct Tree {
    struct Node {
        int l, r, len; // 区间范围
        int sum;       // 区间和
        int tag;       // 懒标记
    } tr[N << 2];

    void pushup(int u) {
        tr[u].sum = tr[ls].sum + tr[rs].sum; // 汇总区间
    }

    // 修改u节点的懒标记和统计信息
    void change(int u, int tag) {
        tr[u].tag = tag;
        if (tr[u].tag == 1) tr[u].sum = tr[u].len;
        if (tr[u].tag == 0) tr[u].sum = 0;
    }

    void pushdown(int u) {
        if (~tr[u].tag) {
            // 向左儿子传递
            change(ls, tr[u].tag);
            // 向左儿子传递
            change(rs, tr[u].tag);
            // 终于完成向左右儿子传递懒标记的任务，将自己的懒标记清除
            tr[u].tag = -1;
        }
    }

    void build(int u, int l, int r) {
        tr[u].l = l, tr[u].r = r, tr[u].len = r - l + 1; // 整体区间[1,n]，管辖区间[l,r]
        tr[u].tag = -1;
        if (l == r) return;
        build(ls, l, mid);
        build(rs, mid + 1, r);
    }

    void modify(int u, int L, int R, int v) {
        if (tr[u].tag == v) return; // 剪枝,否则无法通过12测试点
        // 区间都修改为统一的值，如果上一次你修改成了1，本次还是修改成了1，那么只留同样的就可以了，不用再重新标识

        // 完整命中，修改自己的懒标记和统计信息
        int l = tr[u].l, r = tr[u].r;
        if (l > R || r < L) return;
        if (l >= L && r <= R) {
            change(u, v);
            return;
        }
        pushdown(u);
        modify(ls, L, R, v), modify(rs, L, R, v);
        pushup(u);
    }

    int query(int u, int L, int R) {
        if (tr[u].sum == 0) return 0; // 整体加一块都是0，那你想查找子区间，也肯定是0，剪枝,不加这句，第12个测试点TLE

        int l = tr[u].l, r = tr[u].r;
        if (l > R || r < L) return 0;
        if (l >= L && r <= R) return tr[u].sum; // 如果完整命中, 返回现成的

        pushdown(u);
        return query(ls, L, R) + query(rs, L, R);
    }
} T[92]; // A~Z:65+26=91,所以开92

int main() {
#ifndef ONLINE_JUDGE
    freopen("P2787.in", "r", stdin);
#endif
    // 加快读入
    ios::sync_with_stdio(false), cin.tie(0);

    cin >> n >> m;                                                  // 字母个数，操作次数
    for (int i = 1; i <= n; i++) cin >> a[i], a[i] = toupper(a[i]); // 读入每个字符，并且转成大写

    for (char i = 'A'; i <= 'Z'; i++) T[i].build(1, 1, n); // 创建26个线段树
    for (int i = 1; i <= n; i++) T[a[i]].modify(1, i, i, 1);
    // 枚举字符串中每个字符，两个属性：哪根树，哪个位置，修改为1

    while (m--) {
        int op, l, r;
        char k;
        cin >> op >> l >> r;
        if (op == 1) {
            cin >> k;
            k = toupper(k);
            cout << T[k].query(1, l, r) << endl; // 查询第k棵线段树中[x,y]之间的区间和
        }
        if (op == 2) {
            cin >> k;
            k = toupper(k);
            for (char i = 'A'; i <= 'Z'; i++) // 枚举每棵线段树
                if (i == k)
                    T[i].modify(1, l, r, 1); // 修改的是第i根线段树，区间整体修改为1
                else
                    T[i].modify(1, l, r, 0); // 其它线段树，区间整体修改为0
        }

        if (op == 3) {
            for (char i = 'A'; i <= 'Z'; i++) {
                f[i] = T[i].query(1, l, r); // 将第i棵线段树中[x,y]的区间和保存起来，放到f[i]中
                T[i].modify(1, l, r, 0);    // 放完了就修改为0,表示清除掉懒标记
            }

            // 利用区间修改和桶，完成类似于排序的操作
            for (char i = 'A'; i <= 'Z'; i++)
                if (f[i]) T[i].modify(1, l, l + f[i] - 1, 1), l += f[i];
        }
    }
    return 0;
}