python/TangDou/AcWing/SegmentTree/P1253_ODT.cpp

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
#define int long long // 不开LONG LONG见祖宗
int n, m;

// TLE 掉最后的两个测试点
// 柯朵莉树模板
// 不要采用构建函数与{}混搭的使用，WA的让你怀疑人生！
// 以后记住：在声明结构体时，不要使用构造函数!!
struct Node {
    int l, r;      // l和r表示这一段的起点和终点
    mutable int v; // v表示这一段上所有元素相同的值是多少,注意关键字 mutable,使得set中结构体属性可修改
    bool operator<(const Node &b) const {
        return l < b.l; // 规定按照每段的左端点排序
    }
};
set<Node> s; // 柯朵莉树的区间集合

// 分裂:[l,x-1],[x,r]
set<Node>::iterator split(int x) {
    auto it = s.lower_bound({x});
    if (it != s.end() && it->l == x) return it; // 一击命中
    it--;                                       // 没有找到就减1个继续找
    if (it->r < x) return s.end();              // 真的没找到,返回s.end()

    int l = it->l, r = it->r, v = it->v; // 没有被返回，说明找到了,记录下来，防止后面删除时被破坏
    s.erase(it);                         // 删除整个区间
    s.insert({l, x - 1, v});             //[l,x-1]拆分
    // insert函数返回pair，其中的first是新插入结点的迭代器
    return s.insert({x, r, v}).first; //[x,r]拆分
}

// 区间加
void add(int l, int r, int v) {
    // 必须先计算itr,后计算itl
    auto R = split(r + 1), L = split(l);
    for (auto it = L; it != R; it++) it->v += v;
}
// 区间赋值
void assign(int l, int r, int v) {
    auto R = split(r + 1), L = split(l);
    s.erase(L, R);       // 删除旧区间
    s.insert({l, r, v}); // 增加新区间
}

int query(int l, int r) {
    int res = -LONG_LONG_MAX; // 这个最小值太BT了，需要写成这么小！
    auto R = split(r + 1), L = split(l);
    for (; L != R; L++) res = max(res, L->v);
    return res;
}
/*
答案：
7
6
-1
*/
signed main() {
#ifndef ONLINE_JUDGE
    freopen("P1253.in", "r", stdin);
#endif
    // 加快读入
    ios::sync_with_stdio(false), cin.tie(0);

    // 柯朵莉初始化
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        int x;
        cin >> x;
        s.insert({i, i, x}); // ① 需要初始化多个独立的区间
    }

    while (m--) {
        int op;
        int l, r, x;
        cin >> op;
        cin >> l >> r;
        if (op == 1) {
            cin >> x;
            assign(l, r, x); // ②平推
        } else if (op == 2) {
            cin >> x;
            add(l, r, x); // ③区间内都加上x
        } else
            printf("%lld\n", query(l, r)); // ④查询
    }
    return 0;
}