[TJOI2009]开关【线段树】

[TJOI2009]开关

【线段树】
重点在于$\text{pushdown}$操作，注意到如果原来已经有了开关一次的标记，那么再次标记这个区间时就相当于消除原来这个标记，否则对这个节点进行标记。

#include <bits/stdc++.h>
using namespace std;
const int N=(int)1e5+50;
int n,m;
inline int read(){
	int cnt=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-') f=-f;c=getchar();}
	while(isdigit(c)){cnt=(cnt<<1)+(cnt<<3)+(c^48);c=getchar();}
	return cnt*f;
}
struct tree{int sum,l,r,tag;}tr[N<<2];
inline void pushup(int p){tr[p].sum=tr[p<<1].sum+tr[p<<1|1].sum;}
inline void pushdown(int p){
	if(tr[p].tag==0) return;
	tr[p<<1].sum=tr[p<<1].r-tr[p<<1].l+1-tr[p<<1].sum;
	tr[p<<1|1].sum=tr[p<<1|1].r-tr[p<<1|1].l+1-tr[p<<1|1].sum;
	if(tr[p<<1].tag==1) tr[p<<1].tag=0;
	else tr[p<<1].tag=1;
	if(tr[p<<1|1].tag==1) tr[p<<1|1].tag=0;
	else tr[p<<1|1].tag=1;
	tr[p].tag=0;
}
void build(int p,int l,int r){
	tr[p].l=l;tr[p].r=r;
	if(l==r) {tr[p].sum=0;return;}
	int mid=(l+r)>>1;
	build(p<<1,l,mid);build(p<<1|1,mid+1,r);
	pushup(p); 
}
int query(int p,int l,int r){
	if(l<=tr[p].l&&r>=tr[p].r) return tr[p].sum;
	pushdown(p);int val=0;
	int mid=(tr[p].l+tr[p].r)>>1;
	if(l<=mid) val+=query(p<<1,l,r);
	if(r>mid) val+=query(p<<1|1,l,r);
	return val;
}
void modify(int p,int l,int r){
	if(l<=tr[p].l&&r>=tr[p].r){
		tr[p].sum=(tr[p].r-tr[p].l+1-tr[p].sum);
		if(tr[p].tag==1) tr[p].tag=0;
		else tr[p].tag=1;
		return;
	}
	pushdown(p);
	int mid=(tr[p].l+tr[p].r)>>1;
	if(l<=mid) modify(p<<1,l,r);
	if(r>mid) modify(p<<1|1,l,r);
	pushup(p);
}
signed main(){
	int c,a,b,ans;
	n=read(),m=read();build(1,1,n);
	for(int i=1;i<=m;++i){
		c=read(),a=read(),b=read();
		if(c==0) modify(1,a,b);
		if(c==1) ans=query(1,a,b),printf("%d\n",ans);
	}
	return 0;
}