bzoj1858: [Scoi2010]序列操作

很早之前就做了

线段树裸题

据说可以只维护一段区间是否全是1或者0，复杂度应该没有保证？也不觉得更好写一些。。

 

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>

using namespace std;

void setIO(const string& a) {
    freopen((a+".in").c_str(), "r", stdin);
    freopen((a+".out").c_str(), "w", stdout);
}

struct Data {
    int s[2], l[2], r[2], m[2];
    Data() {
        memset(s, 4*(sizeof s), 0);
    }
};

int n;

Data merge(const Data& lhs, const Data& rhs) {
    Data res;
    
    int n1 = lhs.s[0] + lhs.s[1];
    int n2 = rhs.s[0] + rhs.s[1];
    
    for(int i = 0; i < 2; i++) {
        res.s[i] = lhs.s[i] + rhs.s[i];
        res.l[i] = lhs.l[i];
        if(lhs.l[i] == n1) res.l[i] += rhs.l[i];
        res.r[i] = rhs.r[i];
        if(rhs.r[i] == n2) res.r[i] += lhs.r[i];
        res.m[i] = max(max(lhs.m[i], rhs.m[i]), lhs.r[i] + rhs.l[i]);
    }
    
    return res;
}

const int N = 100000 + 10;

int a[N];


struct SegmentTree {
    Data da[N * 4];    
    int tag[N * 4];// -1 : none  2 : rever
    int lft, rgt, w;
    
    void add_tag(int s, int l, int r, int w) {
        if(w == -1) return;
        int len = r - l + 1;
        if(w != 2) {
            tag[s] = w;
            da[s].s[w] = da[s].m[w] = da[s].l[w] = da[s].r[w] = len;
            w ^= 1;
            da[s].s[w] = da[s].m[w] = da[s].l[w] = da[s].r[w] = 0;
        }else {
            swap(da[s].s[0], da[s].s[1]);
            swap(da[s].l[0], da[s].l[1]);
            swap(da[s].r[0], da[s].r[1]);
            swap(da[s].m[0], da[s].m[1]);
            if(tag[s] == -1) tag[s] = w;
            else if(tag[s] == 2) tag[s] = -1;
            else tag[s] ^= 1;
        }
    }
    
    #define mid ((l + r) >> 1)
    #define ls s << 1, l, mid
    #define rs s << 1 | 1, mid + 1, r
    
    void down(int s, int l, int r) {
        if(tag[s] == -1) return;
        add_tag(ls, tag[s]);
        add_tag(rs, tag[s]);
        tag[s] = -1;
    }
    
    void build(int s, int l, int r) {
        tag[s] = -1;
        if(l == r) {
            scanf("%d", &w);
            da[s].s[w] = da[s].l[w] = da[s].r[w] = da[s].m[w] = 1;
            w ^= 1;
            da[s].s[w] = da[s].l[w] = da[s].r[w] = da[s].m[w] = 0;
            return;
        }
        build(ls);
        build(rs);
        da[s] = merge(da[s << 1], da[s << 1 | 1]);
    }
    
    Data query(int s, int l, int r) {
        if(lft <= l && r <= rgt) return da[s];
        down(s, l, r);
        if(rgt <= mid) return query(ls);
        if(mid < lft) return query(rs);
        return merge(query(ls), query(rs));
    }
    
    void modify(int s, int l, int r) {
        if(lft <= l && r <= rgt) {
            add_tag(s, l, r, w);
            return;
        }
        down(s, l, r);
        if(lft <= mid) modify(ls);
        if(mid < rgt) modify(rs);
        da[s] = merge(da[s << 1], da[s << 1 | 1]);
    }
    
    #undef mid
    #undef ls
    #undef rs
    
    Data Query(int l, int r) {
        lft = l, rgt = r;
        return query(1, 1, n);
    }
    
    void Modify(int l, int r, int w) {
        lft = l, rgt = r, this->w = w;
        modify(1, 1, n);
    }
    
}seg;

int main() {
    int m, opt, l, r;
    scanf("%d%d", &n, &m);
    seg.build(1, 1, n);
//    cerr << seg.Query(1, 6).s[1] << endl;
    while(m--) {
        scanf("%d%d%d", &opt, &l, &r); ++l, ++r;
        if(opt <= 2) seg.Modify(l, r, opt);
        else if(opt == 3) printf("%d\n", seg.Query(l, r).s[1]);
        else printf("%d\n", seg.Query(l, r).m[1]);
    }
    
    return 0;
}