洛谷—P3870 開關(線段樹入門)

在這裏插入圖片描述
此處應該注意lazy的維護,即lazy的值應該是在1-0之間翻轉的,其他沒問題

代碼:

#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
#define ll long long
ll sum[maxn<<2],tag[maxn<<2];
void pushdown(int p,int l,int r)
{
	if(tag[p])
	{
		int mid=(l+r)>>1;
		tag[p<<1]^=1;
		tag[p<<1|1]^=1;
		sum[p<<1]=(mid-l+1)-sum[p<<1];
		sum[p<<1|1]=(r-mid)-sum[p<<1|1];
		tag[p]=0;
	} 
}
void pushup(int p,int l,int r)
{
	sum[p]=sum[p<<1]+sum[p<<1|1];
}
void change(int p,int l,int r,int ql,int qr)
{
	if(ql<=l&&qr>=r)
	{
		tag[p]^=1;
		sum[p]=(r-l+1)-sum[p];
		return;
	}
	pushdown(p,l,r);
	int mid=(l+r)>>1;
	if(ql<=mid) change(p<<1,l,mid,ql,qr);
	if(qr>mid) change(p<<1|1,mid+1,r,ql,qr);
	pushup(p,l,r);
}
ll query(int p,int l,int r,int ql,int qr)
{
	if(ql<=l&&qr>=r) return sum[p];
	pushdown(p,l,r);
	int mid=(l+r)>>1;
	ll res=0;
	if(ql<=mid) res+=query(p<<1,l,mid,ql,qr);
	if(qr>mid) res+=query(p<<1|1,mid+1,r,ql,qr);
	return res;
}
int main()
{
	int n,m;
	cin>>n>>m;
	while(m--)
	{
		int temp,x,y;
		cin>>temp>>x>>y;
		if(temp==0) change(1,1,n,x,y);
		else cout<<query(1,1,n,x,y)<<endl;
	}
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章