[CodeVS 4927] 线段树练习5：两个Lazy Tag的线段树

题意：维护一个长为n（n<=100000）的数列，支持区间加减、区间赋值两种修改，区间最小值、区间最大值、区间求和三种查询。

发现自己记不清线段树区间赋值怎么写了，而且感觉自己的写法和刘汝佳老师的蓝书上的代码不一样，于是做一做这道题。

刘汝佳老师定义赋值标记的优先级高于累加标记，于是同时有两个标记的时候需要先考虑赋值标记，再加上累加标记的值。我的写法略有不同。和CPU监控中的分析一样，加法、赋值混合运算，无论以何种顺序出现，均可化为零或一个加法、零或一个赋值。赋值标记覆盖累加标记，如果已经有赋值标记，再做加法，直接加到赋值标记上。赋值标记用一个布尔值即可，表示是否将min/max下传。

我把两种修改写一起，三种查询写一起，堆式存储。写在一起，代码会短，常数随之略有增加。

#include <cstdio>#include <cctype>#include <algorithm>#define ALL 1, 1, nusing namespace std;typedef long long ll;const int MAX_N = 100000;const ll inf = 1LL<<61;ll x[MAX_N+1];struct Node {    ll mn, mx, sum, a;    bool b;};struct Segment_Tree {    enum op {        ADD, SET    };    Node t[4*MAX_N];    void maintain(int o)    {        Node& self = t[o], & lc = t[o*2], & rc = t[o*2+1];        self.mn = min(lc.mn, rc.mn);        self.mx = max(lc.mx, rc.mx);        self.sum = lc.sum + rc.sum;    }    void pushdown(Node u, Node& v, int w)    {        ll x = u.mn, a = u.a;        // SET        if (u.b)            v = (Node){x, x, x*w, 0, true};        // ADD        else if (v.b)            v = (Node){v.mn+a, v.mx+a, v.sum+w*a, 0, true};        else            v = (Node){v.mn+a, v.mx+a, v.sum+w*a, v.a+a, false};    }    void pushdown(int o, int w)    {        pushdown(t[o], t[o*2], (w+1)/2);        pushdown(t[o], t[o*2+1], w/2);        t[o].a = 0;        t[o].b = false;    }    void build(ll A[], int o, int l, int r)    {        if (l == r) {            t[o] = (Node){A[l], A[l], A[l], 0, false};            return;        }        int m = (l+r)/2;        build(A, o*2, l, m);        build(A, o*2+1, m+1, r);        maintain(o);    }       void modify(int x, int y, ll v, op c, int o, int l, int r)    {        if (x <= l && r <= y) {            if (c == ADD)                pushdown((Node){0, 0, 0, v, false}, t[o], r-l+1);            else                pushdown((Node){v, 0, 0, 0, true}, t[o], r-l+1);            return;        }        pushdown(o, r-l+1);        int m = (l+r)/2;        if (x <= m)            modify(x, y, v, c, o*2, l, m);        if (y > m)            modify(x, y, v, c, o*2+1, m+1, r);        maintain(o);    }    void query(int x, int y, Node& d, int o, int l, int r)    {        if (x <= l && r <= y) {            d.mn = min(d.mn, t[o].mn);            d.mx = max(d.mx, t[o].mx);            d.sum += t[o].sum;            return;        }        pushdown(o, r-l+1);        int m = (l+r)/2;        if (x <= m)            query(x, y, d, o*2, l, m);        if (y > m)            query(x, y, d, o*2+1, m+1, r);        maintain(o);    }} T;template<typename T>inline void read(T& x){    char c = getchar();    x = 0;    int s = 1;    while (!isdigit(c)) {        if (c == '-')            s = -1;        c = getchar();    }    while (isdigit(c)) {        x = x*10 + c - '0';        c = getchar();    }    x *= s;}int main(){    int n, m;    read(n), read(m);    for (int i = 1; i <= n; ++i)        read(x[i]);    T.build(x, ALL);    for (int i = 1; i <= m; ++i) {        char s[4];        ll x, y, z;        scanf("%s", s), read(x), read(y);        if (s[0] == 'a') {            read(z);            T.modify(x, y, z, Segment_Tree::ADD, ALL);        } else if (s[0] == 's' && s[1] == 'e') {            read(z);            T.modify(x, y, z, Segment_Tree::SET, ALL);        } else {            Node r = (Node){inf, -inf, 0, 0, false};            T.query(x, y, r, ALL);            printf("%lld\n", s[0] == 's' ? r.sum : (s[1] == 'i' ? r.mn : r.mx));        }    }    return 0;}