[BZOJ1230][Usaco2008 Nov]lites 开关灯

1230: [Usaco2008 Nov]lites 开关灯

Time Limit: 10 Sec  Memory Limit: 162 MB Submit: 1606  Solved: 844 [Submit][Status][Discuss]

Description

Farmer John尝试通过和奶牛们玩益智玩具来保持他的奶牛们思维敏捷. 其中一个大型玩具是牛栏中的灯. N (2 <= N <= 100,000) 头奶牛中的每一头被连续的编号为1..N, 站在一个彩色的灯下面.刚到傍晚的时候, 所有的灯都是关闭的. 奶牛们通过N个按钮来控制灯的开关; 按第i个按钮可以改变第i个灯的状态.奶牛们执行M (1 <= M <= 100,000)条指令, 每个指令都是两个整数中的一个(0 <= 指令号 <= 1). 第1种指令(用0表示)包含两个数字S_i和E_i (1 <= S_i <= E_i <= N), 它们表示起始开关和终止开关. 奶牛们只需要把从S_i到E_i之间的按钮都按一次, 就可以完成这个指令. 第2种指令(用1表示)同样包含两个数字S_i和E_i (1 <= S_i <= E_i <= N), 不过这种指令是询问从S_i到E_i之间的灯有多少是亮着的. 帮助FJ确保他的奶牛们可以得到正确的答案.

Input

* 第 1 行: 用空格隔开的两个整数N和M

 * 第 2..M+1 行: 每行表示一个操作, 有三个用空格分开的整数: 指令号, S_i 和 E_i

Output

第 1..询问的次数 行: 对于每一次询问, 输出询问的结果.

Sample Input

4 5
0 1 2
0 2 4
1 2 3
0 2 4
1 1 4

输入解释:
一共有4盏灯; 5个指令. 下面是执行的情况:
        灯
            1 2 3 4
  Init:     O O O O   O = 关  * = 开
  0 1 2 ->  * * O O  改变灯 1 和 2 的状态
  0 2 4 ->  * O * *
  1 2 3 ->  1        输出在2..3的范围内有多少灯是亮的
  0 2 4 ->  * * O O  改变灯 2 ,3 和 4 的状态
  1 1 4 ->  2        输出在1..4的范围内有多少灯是亮的

Sample Output

1
2

线段树，维护区间1的个数以及标记

#include <cstdio>
#include <cstring>
char buf[10000000], *ptr = buf - 1;
inline int readint(){
    int n = 0;
    char ch = *++ptr;
    while(ch < '0' || ch > '9') ch = *++ptr;
    while(ch <= '9' && ch >= '0'){
        n = (n << 1) + (n << 3) + ch - '0';
        ch = *++ptr;
    }
    return n;
}
const int maxn = 100000 + 10;
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1 
int sum[maxn << 2] = {0}, col[maxn << 2] = {0};
void inline PushUp(int rt){
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void inline PushDown(int rt, int len){
    if(col[rt]){
        col[rt << 1] = (col[rt << 1] + col[rt]) & 1;
        sum[rt << 1] = len - (len >> 1) - sum[rt << 1];    
        col[rt << 1 | 1] = (col[rt << 1 | 1] + col[rt]) & 1;
        sum[rt << 1 | 1] = (len >> 1) - sum[rt << 1 | 1];
        col[rt] = 0;
    }
}
int Query(int ql, int qr, int l, int r, int rt){
    if(ql <= l && r <= qr) return sum[rt];
    else{
        PushDown(rt, r - l + 1);
        int mid = l + r >> 1, ret = 0;
        if(ql <= mid) ret += Query(ql, qr, lson);
        if(qr > mid) ret += Query(ql, qr, rson);
        return ret;
    }
}
void Update(int ql, int qr, int l, int r, int rt){
    if(ql <= l && r <= qr){
        sum[rt] = r - l + 1 - sum[rt];
        col[rt] = (col[rt] + 1) & 1;
    }
    else{
        PushDown(rt, r - l + 1);
        int mid = l + r >> 1;
        if(ql <= mid) Update(ql, qr, lson);
        if(qr > mid) Update(ql, qr, rson);
        PushUp(rt);
    }
}
int n, m;
int main(){
    fread(buf, sizeof(char), sizeof(buf), stdin);
    n = readint();
    m = readint();
    int opt, s, e;
    while(m--){
        opt = readint();
        s = readint();
        e = readint();
        if(opt == 0) Update(s, e, 1, n, 1);
        else printf("%d\n", Query(s, e, 1, n, 1));
    }
    return 0;
}