BZOJ 1230--lites 开关灯(线段树)

1230: [Usaco2008 Nov]lites 开关灯

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1682  Solved: 876
[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

题目链接：

　　　　http://www.lydsy.com/JudgeOnline/problem.php?id=1230

Solution

　　　　线段树模板题，对一个区间取反时做标记。。。

　　　　都是线段树的基本操作。。。

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
using namespace std;
inline int Read_(){
	int x=0,f=1;char ch=getchar();
	while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
	while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
	return x*f;
}
int n,m,ans;
struct node{
	int l,r,s;
	bool c;
}e[400000];
void build(int l,int r,int num){
	e[num].l=l;e[num].r=r;e[num].s=0;e[num].c=0;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(l,mid,num<<1);
	build(mid+1,r,num<<1|1);
	return;
}
void down_(int num){
	e[num].c=0;
	e[num<<1].c=1-e[num<<1].c;
	e[num<<1].s=e[num<<1].r-e[num<<1].l+1-e[num<<1].s;
	e[num<<1|1].c=1-e[num<<1|1].c;
	e[num<<1|1].s=e[num<<1|1].r-e[num<<1|1].l+1-e[num<<1|1].s;
	return;
}
void update_(int l,int r,int num){
	if(e[num].l==l&&e[num].r==r){
		e[num].c=1-e[num].c;
		e[num].s=e[num].r-e[num].l+1-e[num].s;
		return;
	}
	if(e[num].c) down_(num);
	if(e[num<<1].r>=r) update_(l,r,num<<1);
	else if(e[num<<1].r<l) update_(l,r,num<<1|1);
	else{
		update_(l,e[num<<1].r,num<<1);
		update_(e[num<<1|1].l,r,num<<1|1);
	}
	e[num].s=e[num<<1].s+e[num<<1|1].s;
	return;
}
void solve_(int l,int r,int num){
	if(e[num].l==l&&e[num].r==r){
		ans+=e[num].s;
		return;
	}
	if(e[num].c)down_(num);
	if(e[num<<1].r>=r) solve_(l,r,num<<1);
	else if(e[num<<1].r<l) solve_(l,r,num<<1|1);
	else{
		solve_(l,e[num<<1].r,num<<1);
		solve_(e[num<<1|1].l,r,num<<1|1);
	}
	return;
}
int main(){
	int ty,L,R;
	n=Read_();m=Read_();
	build(1,n,1);
	for(int i=1;i<=m;i++){
		ty=Read_();L=Read_();R=Read_();
		if(!ty) update_(L,R,1);
		else {
			ans=0;
			solve_(L,R,1);
			printf("%d\n",ans);
		}
	}
	return 0;
}

This passage is made by Iscream-2001.