P2572 [SCOI2010] 序列操作 （线段树）

P2572

又是一道颠覆我对线段树认知的题......

线段树每个节点维护八个信息：0/1个数，从左/右起0/1的个数，最长连续的0/1个数。

我们用线段树维护信息是需要区间合并的，因此为保证能够合并，我们才维护以上的八个信息。

打上两个标记：tg1={-1， 0， 1}，-1表示没有，0表示将区间全部变成0，1表示将区间全部变成1；

tg2={0， 1}，表示区间取反。

先上代码：

#include <bits/stdc++.h>
#define mid ((l + r) >> 1)
#define lc k << 1
#define rc k << 1 | 1
#define lson lc, l, mid
#define rson rc, mid + 1, r
using namespace std;
int n, q, a[100001];
struct node {
    int w, b, lw, lb, rw, rb, mw, mb;
    //构造函数，方便复制 
    node(int w = 0, int b = 0, int lw = 0, int lb = 0, int rw = 0, int rb = 0, int mw = 0, int mb = 0) :
    w(w), b(b), lw(lw), lb(lb), rw(rw), rb(rb), mw(mw), mb(mb){}    
}dat[262144];//维护复杂信息多使用结构体 
node merge(node i, node j) {//合并 
    return node(
        i.w + j.w, i.b + j.b,
        (i.b ? i.lw : i.w + j.lw), (i.w ? i.lb : i.b + j.lb),
        (j.b ? j.rw : j.w + i.rw), (j.w ? j.rb : j.b + i.rb),
        max(max(i.mw, j.mw), i.rw + j.lw),
        max(max(i.mb, j.mb), i.rb + j.lb));
}
int len[262144], tg1[262144], tg2[262144];
void build(int k, int l, int r) {
    len[k] = r - l + 1, tg1[k] = -1;
    if (l == r) {int t = a[l]; dat[k] = node(t, t ^ 1, t, t ^ 1, t, t ^ 1, t, t ^ 1); return ;}
    build(lson), build(rson);
    dat[k] = merge(dat[lc], dat[rc]);
}
void P(int k, int typ) {
    node &t = dat[k];
    //区间赋值为0 
    if (typ == 0) tg2[k] = 0, tg1[k] = 0, t = node(0, len[k], 0, len[k], 0, len[k], 0, len[k]);
    //区间赋值为1
    if (typ == 1) tg2[k] = 0, tg1[k] = 1, t = node(len[k], 0, len[k], 0, len[k], 0, len[k], 0);
    //区间取反
    if (typ == 2) tg2[k] ^= 1, swap(t.w, t.b), swap(t.lw, t.lb), swap(t.rw, t.rb), swap(t.mw, t.mb);
}
void pd(int k) {//下传标记 
    if (~tg1[k]) P(lc, tg1[k]), P(rc, tg1[k]);
    if (tg2[k]) P(lc, 2), P(rc, 2);
    tg1[k] = -1, tg2[k] = 0;
}
void Mdf(int k, int l, int r, int L, int R, int t) {
    if (R < l || L > r) return ;//!!!!!!!
    if (L <= l && R >= r) {P(k, t); return ;}
    pd(k);
    Mdf(lson, L, R, t), Mdf(rson, L, R, t);
    dat[k] = merge(dat[lc], dat[rc]);
}
node Qur(int k, int l, int r, int L, int R) {
    if (R < l || L > r) return node();//!!!!!!!
    if (L <= l && R >= r) return dat[k];
    pd(k);
    return merge(Qur(lson, L, R), Qur(rson, L, R));
}
int main() {
    scanf("%d %d", &n, &q);
    for (int i = 1; i <= n; i ++) scanf("%d", a + i);
    build(1, 1, n);
    while (q --) {
        int opt, l, r;
        scanf("%d %d %d", &opt, &l, &r); l ++, r ++;
        if (opt < 3) Mdf(1, 1, n, l, r, opt);
        else {node t = Qur(1, 1, n, l, r); printf("%d\n", opt == 3 ? t.w : t.mw);}
    }
    return 0;
}

用结构体的构造函数可以大大减少代码量，建议维护复杂信息时都采用这种方式。

另外，代码第45,52行的操作加上最好（我没加上就一直RE......）。

巨佬博客

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但是未经作者本人同意，转载文章之后必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。