洛谷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;
}