BZOJ 1230: [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

HINT

Source

线段树。。。。。。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define MAX 100010
using namespace std;
struct tree{
   int l,r,sum,lazy;
}f[MAX*4];
 
void pushup(int i){
    f[i].sum=(f[i*2+1].sum+f[i*2].sum);
}
 
void update(int i,int x)
{
    f[i].sum=(f[i].r-f[i].l+1)-f[i].sum;
    f[i].lazy+=x;
    return;
}
 
void pushdown(int i){
    if(f[i].lazy!=0){
    	if(f[i].lazy%2!=0){
    		update(i*2,f[i].lazy);
        	update(i*2+1,f[i].lazy);
        	f[i].lazy=0;
        	return;
		}
        else{
        	f[i*2].lazy+=f[i].lazy;
        	f[i*2+1].lazy+=f[i].lazy;
        	f[i].lazy=0;
        	return;
		}
    }
    else return;
}
 
void build(int i,int left,int right){
    int mid=(left+right)/2;
    f[i].lazy=0;f[i].l=left;f[i].r=right;
    if(left==right){
        f[i].sum=0;
        return;
    }
    build(i*2,left,mid);
    build(i*2+1,mid+1,right);
    pushup(i);
}
 
void add(int i,int left,int right){
    int mid=(f[i].l+f[i].r)/2;
    if(f[i].l==left&&f[i].r==right){
        update(i,1);
        return;
    }
    pushdown(i);
    if(mid>=right)add(i*2,left,right);  
    else if(mid<left)add(i*2+1,left,right);
    else add(i*2,left,mid),add(i*2+1,mid+1,right);
    pushup(i);
}

int query(int i,int left,int right) {
        int mid=(f[i].l+f[i].r)/2;
        if(f[i].l==left&&f[i].r==right) return f[i].sum;//AK爷别打我
        pushdown(i);
        if(mid>=right) return query(i*2,left,right);
        if(mid<left) return query(i*2+1,left,right);
        return (query(i*2,left,mid)+query(i*2+1,mid+1,right));
}

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    build(1,1,n);
    
    for(int i=1;i<=k;i++){
        int x,y,z;
        scanf("%d",&x);
        if(x==0){
            scanf("%d%d",&y,&z);
            add(1,y,z);
        }
        if(x==1){
            scanf("%d%d",&y,&z);
            printf("%d\n",query(1,y,z));
        }
    }
    return 0;
}

 

此条目发表在BZOJ, USACO, 线段树分类目录。将固定链接加入收藏夹。