线段树-开关

【数据结构2-2】线段树 - 题单 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
int n,m;
const int N=100010;
struct node{
	int l,r,len; //区间长度
	int sum,lazy;//亮的个数，懒标记
};
node tr[N*4];
void pushup(int u){
	tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
}
void build(int u,int l,int r){
	tr[u]={l,r,r-l+1};
	if(l==r){
		tr[u].sum=0;
		tr[u].lazy=0;
		return;
	}	
	int mid=(l+r)/2;
	build(u*2,l,mid);
	build(u*2+1,mid+1,r);
	pushup(u);//更新父亲
}
void down(int u,int x){//修改儿子
	tr[u].sum=tr[u].len-tr[u].sum;
	tr[u].lazy^=1;
}
void pushdown(int u){//如果自己有标记则更新儿子
	if(tr[u].lazy==0) return;
	down(u*2,tr[u].lazy);
	down(u*2+1,tr[u].lazy);
	tr[u].lazy=0;
}
/*
修改操作，搜索到根节点修改，然后需要回溯修改所有父亲
*/
void change(int u,int l,int r){
	if(l<=tr[u].l&&tr[u].r<=r){//找到根节点
	tr[u].sum=tr[u].len-tr[u].sum;//改变
	tr[u].lazy^=1;
		return; 
	}
	pushdown(u);//更新儿子
	int mid=(tr[u].l+tr[u].r)/2;
	if(l<=mid) change(u*2,l,r);
	if(r>mid) change(u*2+1,l,r);
	pushup(u);	//更新父亲
}
/*
查询操作,所有儿子都是父亲过来的，不需要回溯父亲
*/
int query(int u,int l,int r){
	if(l<=tr[u].l&&tr[u].r<=r) return tr[u].sum;
	int mid=(tr[u].l+tr[u].r)/2;
	int res=0;
	pushdown(u);
	if(l<=mid) res+=query(u*2,l,r);
	if(r>mid) res+=query(u*2+1,l,r);
	return res;
}
void solve()
{
	cin>>n>>m;
	build(1,1,n);
	while(m--){
		int s; cin>>s;
		if(s==0){
			int l,r;	
			cin>>l>>r;
			change(1,l,r);
		}else{
			int l,r;
			cin>>l>>r;
			cout<<query(1,l,r)<<endl;
		}
	}
}
signed main()
{
	ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
	solve();
	return 0;
}