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..询问的次数 行: 对于每一次询问, 输出询问的结果.
维护一个区间,支持区间异或和区间查询。
我们维护一棵线段树,维护一个懒标记,当懒标记为1时,下传懒标记,即将子结点的懒标记异或1,再修改一下子结点的权值。
修改子结点权值时,由于修改操作是将1变成0,0变成1,所以修改后的值应该是区间长度减去原来的权值。
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#define lc root<<1,l,mid
#define rc root<<1|1,mid+1,r
#define maxn 200010
using namespace std;
int a[maxn<<2],tag[maxn<<2],aa[maxn];
char s[maxn];
void pushup(int root)
{
a[root]=a[root<<1]+a[root<<1|1];
}
void build(int root,int l,int r)
{
if(l==r){
a[root]=0;
tag[root]=0;
return;
}
int mid=(l+r)>>1;
build(lc);
build(rc);
pushup(root);
}
void pushdown(int root,int l,int r)
{
if(tag[root]){
tag[root<<1]^=1;
tag[root<<1|1]^=1;
int mid=(l+r)>>1;
a[root<<1]=(mid-l+1)-a[root<<1];
a[root<<1|1]=(r-mid)-a[root<<1|1];
tag[root]=0;
}
}
void update(int root,int l,int r,int ul,int ur)
{
if(ul<=l&&ur>=r){
tag[root]^=1;
a[root]=(r-l+1)-a[root];
return;
}
pushdown(root,l,r);
int mid=(l+r)>>1;
if(ul<=mid) update(lc,ul,ur);
if(ur>mid) update(rc,ul,ur);
pushup(root);
}
int query(int root,int l,int r,int ql,int qr)
{
if(ql<=l&&qr>=r) return a[root];
int ans=0;
pushdown(root,l,r);
int mid=(l+r)>>1;
if(ql<=mid) ans+=query(lc,ql,qr);
if(qr>mid) ans+=query(rc,ql,qr);
return ans;
}
int main()
{
int n,m;
cin>>n>>m;
build(1,1,n);
while(m--){
int opt,l,r;
scanf("%d%d%d",&opt,&l,&r);
if(opt==0) update(1,1,n,l,r);
if(opt==1) printf("%d\n",query(1,1,n,l,r));
}
return 0;
}