HDU 3397 Sequence operation (线段树+区间合并+成段更新+多种操作)

题意：有N个为0或为1的数，有M个操作，操作有5种类型。（1）“0 a b”，表示将区间[a,b]范围内的数全部置0。（2）“1 a b”，表示将区间[a,b]内的数全部置1。（3）“2 a b”，表示将区间[a,b]内的数0变成1，1变成0。（4）“3 a b”，表示查询[a,b]范围内1的数。（5）“4 a b”，表示查询[a,b]范围内最长的连续的1。

题解：线段树+区间合并+成段更新
从昨天晚上wa到今天中午，终于a了。

说明一下变量。
$sum$：记录连续数字1。$lsum$：记录左端点开始的连续数字1。$rsum$相反。
$sum0$：记录连续数字0。$lsum0$：记录左端点开始的连续数字0。$rsum0$相反。
$s$：记录数字个数。对应操作4。

操作1、2就是简单的成段更新。
操作4就是区间查询，操作5就是区间合并。这两个查询要分开来写，一个是求最大，一个是求和，细节见代码。

这题关键的就在操作3取反上。我们需要重新开一个懒惰标记，记录取反操作。

因为序列只有0和1，我们只要开辟与记录数字1对应的另外的数组，来记录0即可，遇到取反，只要交换两两的值就好了。

还有比较关键的就是懒惰标记的更新，因为我们不知道原来的标记的状态，所以不能直接赋值，而是要进行取反（与1异或）。

当然做这些还是不够的，因为操作1、2其实就相当于覆盖操作，那么之前的取反操作就不需要了，所以覆盖得在取反之前，这点在PushDown里体现。说的具体一点，如果我们在进行覆盖操作的时候，遇到取反标记，那么没必要往下更新取反标记（因为覆盖了），但是如果是当前有进行取反的操作，我们还是要往下更新。所以，在进行覆盖操作的时候，我们置0的是左右子结点的取反标记，这很重要。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
//区间合并
const int MAXN = 1e5 + 5;
int sum[MAXN << 2], lsum[MAXN << 2], rsum[MAXN << 2], x[MAXN], add[MAXN << 2], s[MAXN << 2], rev[MAXN << 2];
int sum0[MAXN << 2], lsum0[MAXN << 2], rsum0[MAXN << 2], s0[MAXN << 2];
struct Node {
    int l, r;
    int mid() { return (l + r) >> 1; }
} tree[MAXN << 2];

void PushDown(int rt, int m) {
    if (~add[rt]) {
        rev[rt << 1] = rev[rt << 1 | 1] = 0;
        add[rt << 1] = add[rt];
        add[rt << 1 | 1] = add[rt];
        s[rt << 1] = sum[rt << 1] = lsum[rt << 1] = rsum[rt << 1] = add[rt] * (m - (m >> 1));
        s[rt << 1 | 1] = sum[rt << 1 | 1] = lsum[rt << 1 | 1] = rsum[rt << 1 | 1] = add[rt] * (m >> 1);
        s0[rt << 1] = sum0[rt << 1] = lsum0[rt << 1] = rsum0[rt << 1] = (add[rt] ^ 1) * (m - (m >> 1));
        s0[rt << 1 | 1] = sum0[rt << 1 | 1] = lsum0[rt << 1 | 1] = rsum0[rt << 1 | 1] = (add[rt] ^ 1) * (m >> 1);
        add[rt] = -1;
    }
    if (rev[rt]) {
        rev[rt << 1] ^= 1;
        rev[rt << 1 | 1] ^= 1;
        swap(sum[rt << 1], sum0[rt << 1]);
        swap(lsum[rt << 1], lsum0[rt << 1]);
        swap(rsum[rt << 1], rsum0[rt << 1]);
        swap(s[rt << 1], s0[rt << 1]);

        swap(sum[rt << 1 | 1], sum0[rt << 1 | 1]);
        swap(lsum[rt << 1 | 1], lsum0[rt << 1 | 1]);
        swap(rsum[rt << 1 | 1], rsum0[rt << 1 | 1]);
        swap(s[rt << 1 | 1], s0[rt << 1 | 1]);
        rev[rt] = 0;
    }
}
void PushUp(int rt) {
    int m = tree[rt].mid();
    s[rt] = s[rt << 1] + s[rt << 1 | 1];
    sum[rt] = max(max(sum[rt << 1], sum[rt << 1 | 1]), rsum[rt << 1] + lsum[rt << 1 | 1]);
    lsum[rt] = lsum[rt << 1];
    rsum[rt] = rsum[rt << 1 | 1];
    if (lsum[rt] == (m - tree[rt].l + 1)) lsum[rt] += lsum[rt << 1 | 1];
    if (rsum[rt] == (tree[rt].r - m)) rsum[rt] += rsum[rt << 1];

    s0[rt] = s0[rt << 1] + s0[rt << 1 | 1];
    sum0[rt] = max(max(sum0[rt << 1], sum0[rt << 1 | 1]), rsum0[rt << 1] + lsum0[rt << 1 | 1]);
    lsum0[rt] = lsum0[rt << 1];
    rsum0[rt] = rsum0[rt << 1 | 1];
    if (lsum0[rt] == (m - tree[rt].l + 1)) lsum0[rt] += lsum0[rt << 1 | 1];
    if (rsum0[rt] == (tree[rt].r - m)) rsum0[rt] += rsum0[rt << 1];
}
void build(int l, int r, int rt) {
    tree[rt].l = l;
    tree[rt].r = r;
    add[rt] = -1;
    rev[rt] = 0;
    if (l == r) {
        if (x[l]) {
            sum[rt] = lsum[rt] = rsum[rt] = s[rt] = 1;
            sum0[rt] = lsum0[rt] = rsum0[rt] = s0[rt] = 0;
        }
        else {
            sum[rt] = lsum[rt] = rsum[rt] = s[rt] = 0;
            sum0[rt] = lsum0[rt] = rsum0[rt] = s0[rt] = 1;
        }
        return;
    }
    int m = tree[rt].mid();
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
    PushUp(rt);
}
void update(int c, int l, int r, int rt) {
    if (tree[rt].l == l && r == tree[rt].r) {
        if (c == 0) {
            rev[rt] = 0;
            add[rt] = 0;
            sum[rt] = lsum[rt] = rsum[rt] = s[rt] = 0;
            sum0[rt] = lsum0[rt] = rsum0[rt] = s0[rt] = r - l + 1;
        }
        else if (c == 1) {
            rev[rt] = 0;
            add[rt] = 1;
            sum[rt] = lsum[rt] = rsum[rt] = s[rt] = r - l + 1;
            sum0[rt] = lsum0[rt] = rsum0[rt] = s0[rt] = 0;
        }
        else {
            rev[rt] ^= 1;
            swap(sum[rt], sum0[rt]);
            swap(lsum[rt], lsum0[rt]);
            swap(rsum[rt], rsum0[rt]);
            swap(s[rt], s0[rt]);
        }
        return;
    }
    if (tree[rt].l == tree[rt].r) return;
    PushDown(rt, tree[rt].r - tree[rt].l + 1);
    int m = tree[rt].mid();
    if (r <= m) update(c, l, r, rt << 1);
    else if (l > m) update(c, l, r, rt << 1 | 1);
    else {
        update(c, l, m, rt << 1);
        update(c, m + 1, r, rt << 1 | 1);
    }
    PushUp(rt);
}
int query(int id, int l, int r, int rt) {
    if (l == tree[rt].l && r == tree[rt].r) {
        return id == 3 ? s[rt] : sum[rt];
    }
    PushDown(rt, tree[rt].r - tree[rt].l + 1);
    int m = tree[rt].mid();
    int res = 0;
    if (r <= m) res += query(id, l, r, rt << 1);
    else if (l > m) res += query(id, l, r, rt << 1 | 1);
    else {
        int res1 = query(id, l, m, rt << 1);
        int res2 = query(id, m + 1, r, rt << 1 | 1);
        if (id == 3) res = res1 + res2;
        else {
            int x = min(rsum[rt << 1], m - l + 1);  //这里注意，取min
            int y = min(lsum[rt << 1 | 1], r - m);
            res = max(max(res1, res2), x + y);
        }
    }
    return res;
}
int t, n, m, id, a, b;;
int main() {
    //freopen("in.txt", "r", stdin);
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= n; i++) scanf("%d", &x[i]);
        build(1, n, 1);
        while (m--) {
            scanf("%d%d%d", &id, &a, &b);
            if (id <= 2) {
                update(id, a + 1, b + 1, 1);
            }
            else {
                printf("%d\n", query(id, a + 1, b + 1, 1));
            }
        }
    }
    return 0;
}