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的范围内有多少灯是亮的
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
2
Solution
线段树，支持区间异或和区间求和。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
#include<cstdio> #include<iostream> using namespace std; int n,m,tree[800100],lazy[800100]; void push(int id,int x) { if (lazy[id]!=0) { lazy[id*2]^=1; lazy[id*2+1]^=1; tree[id*2]=(x+1)/2-tree[id*2]; tree[id*2+1]=x/2-tree[id*2+1]; lazy[id]=0; } } void change(int id,int l,int r,int x,int y) { if (l>y||r<x) return; push(id,r-l+1); if (x<=l&&r<=y) { lazy[id]^=1;tree[id]=r-l+1-tree[id];return;} int mid=(l+r)/2; change(id*2,l,mid,x,y); change(id*2+1,mid+1,r,x,y); tree[id]=tree[id*2]+tree[id*2+1]; } int query(int id,int l,int r,int x,int y) { if (l>y||r<x) return 0; push(id,r-l+1); if (x<=l&&r<=y) return tree[id]; int mid=(l+r)/2; return query(id*2,l,mid,x,y)+query(id*2+1,mid+1,r,x,y); } int main() { scanf("%d%d",&n,&m); for (int i=1;i<=m;i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); if (x==0) change(1,1,n,y,z);else printf("%d\n",query(1,1,n,y,z)); } return 0; } |
