洛谷2846 USACO08NOV 光开关 Light Switching

题目：Light Switching

思路：区间修改区间查询的线段树。

代码：#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
using namespace std;

#define maxn 100000

int n,m;
int a[4*maxn+5]={0};
int lzy[4*maxn+5]={0};

int p,q;

void push_down(int o,int lc,int rc,int l1,int r1,int l2,int r2){
if(lzy[o]) {
lzy[lc]^=1,lzy[rc]^=1;
a[lc]=r1-l1+1-a[lc],a[rc]=r2-l2+1-a[rc];
lzy[o]=0;
}
}

void update(int o,int l,int r){
if(l>=p&&r<=q) {
lzy[o]^=1,a[o]=r-l+1-a[o];
return ;
}
if(l>q||r<p) return ;

int lc=o*2,rc=o*2+1,mid=l+(r-l)/2;
push_down(o,lc,rc,l,mid,mid+1,r);
update(lc,l,mid),update(rc,mid+1,r);
a[o]=a[lc]+a[rc];
}

int query(int o,int l,int r){
if(l>=p&&r<=q) {
return a[o];
}
if(l>q||r<p) return 0;

int lc=o*2,rc=o*2+1,mid=l+(r-l)/2;
push_down(o,lc,rc,l,mid,mid+1,r);
return query(lc,l,mid)+query(rc,mid+1,r);
}

int main() {
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int opr;
scanf("%d%d%d",&opr,&p,&q);
if(!opr){
update(1,1,n);
} else {
printf("%d\n",query(1,1,n));
}
}

return 0;
}