B1230 [Usaco2008 Nov]lites 开关灯 线段树

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

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
#define duke(i,a,n) for(int i = a;i <= n;i++)
#define lv(i,a,n) for(int i = a;i >= n;i--)
#define clean(a) memset(a,0,sizeof(a))
const int INF = 1 << 30;
typedef long long ll;
typedef double db;
template <class T>
void read(T &x)
{
    char c;
    bool op = 0;
    while(c = getchar(), c < ‘0‘ || c > ‘9‘)
        if(c == ‘-‘) op = 1;
    x = c - ‘0‘;
    while(c = getchar(), c >= ‘0‘ && c <= ‘9‘)
        x = x * 10 + c - ‘0‘;
    if(op) x = -x;
}
template <class T>
void write(T x)
{
    if(x < 0) putchar(‘-‘), x = -x;
    if(x >= 10) write(x / 10);
    putchar(‘0‘ + x % 10);
}
int tree[400010],n,m;
int lazy[400010];
void push_down(int o,int l,int r)
{
    if(lazy[o])
    {
        int mid = (l + r) >> 1;
//        tree[o] = r - l + 1 - tree[o];
        lazy[o << 1] ^= 1;
        tree[o << 1] =  (mid - l + 1) - tree[o << 1];
        lazy[o << 1 | 1] ^= 1;
        tree[o << 1 | 1] = (r - mid) - tree[o << 1 | 1];
        lazy[o] = 0;
    } 
}
void add(int o,int l,int r,int x,int y)
{
    if(l == x && r == y)
    {
        tree[o] = r - l + 1 - tree[o];
        lazy[o] ^= 1;
        return;
    }
    push_down(o,l,r);
    int mid = (l + r) >> 1;
    if(mid >= y)
        add(o << 1,l,mid,x,y);
    else if(mid < x)
        add(o << 1 | 1,mid + 1,r,x,y);
    else
    {
        add(o << 1,l,mid,x,mid);
        add(o << 1 | 1,mid + 1,r,mid + 1,y); 
    }
    tree[o] = tree[o << 1] + tree[o << 1 | 1];
}
int query(int o,int l,int r,int x,int y)
{
    push_down(o,l,r);
    if(l == x && r == y)
        return tree[o];
    int mid = (l + r) >> 1;
    if(mid >= y)
        return query(o << 1,l,mid,x,y);
    else if(mid < x)
        return query(o << 1 | 1,mid + 1,r,x,y);
    else
    {
        return query(o << 1,l,mid,x,mid) + query(o << 1 | 1,mid + 1,r,mid + 1,y);
    }
}
int main()
{
    read(n);read(m);
    duke(i,1,m)
    {
        int k,l,r;
        read(k);read(l);read(r);
        if(k == 0)
        add(1,1,n,l,r);
        else
        printf("%d\n",query(1,1,n,l,r));
    }
    return 0;
}