EOJ 2525/USACO 2008 November Gold Light Switching【線段樹】

題意簡介

---

有n個燈，m次操作，0表示一段區間內燈的狀態全部反轉，1表示詢問一段區間內亮着的燈的數量。

說明

---

裸線段樹……借鑑了模板，算是第一次寫。

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e5+5;

int st[maxn<<2], add[maxn<<2], n, m, flag, s, e;

void pushdown(int rt, int l, int r)
{
    int mid = (l+r) >> 1;
    st[rt<<1] = mid-l+1 - st[rt<<1];
    st[rt<<1|1] = r-mid - st[rt<<1|1];
    add[rt<<1] ^= add[rt];
    add[rt<<1|1] ^= add[rt];
    add[rt] = 0;
}

void update(int rt, int s, int e, int l, int r)
{
    if (s <= l && r <= e)
    {
        st[rt] = r-l+1 - st[rt];
        add[rt] ^= 1;
        return;
    }
    int mid = (l+r) >> 1;
    if (add[rt]) pushdown(rt, l, r);
    if (s <= mid) update(rt<<1, s, e, l, mid);
    if (e > mid) update(rt<<1|1, s, e, mid+1, r);
    st[rt] = st[rt<<1] + st[rt<<1|1];
}

int query(int rt, int s, int e, int l, int r)
{
    if (s <= l && r <= e) return st[rt];
    int ans = 0, mid = (l+r) >> 1;
    if (add[rt]) pushdown(rt, l, r);
    if (s <= mid) ans += query(rt<<1, s, e, l, mid);
    if (e > mid) ans += query(rt<<1|1, s, e, mid+1, r);
    return ans;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 0; i < m; ++i)
    {
        scanf("%d%d%d", &flag, &s, &e);
        if (flag) printf("%d\n", query(1, s, e, 1, n));
        else update(1, s, e, 1, n);
    }
    return 0;
}