0pt初学线段树求调
查看原帖
0pt初学线段树求调
469213
SugarKite楼主2023/11/4 16:28
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m;
int a[5000050],tre[5000050],tag[5000050];
int ls(int x){return 2*x;}
int rs(int x){return 2*x+1;}
void push_up(int p){
	tre[p]=tre[ls(p)]+tre[rs(p)];
}
void add_tag(int p,int pl,int pr){
	tre[p]=(pr-pl+1)-tre[p];
	tag[p]++;
}
void push_down(int p,int pl,int pr){
	if(tag[p]){
		int mid=(pl+pr)>>1;
		add_tag(ls(p),pl,mid);
		add_tag(rs(p),mid+1,pr);
		tag[p]=0;
	}
}
void updata(int p,int L,int R,int pl,int pr){
	if(pl>=L&&pr<=R){
		add_tag(p,pl,pr);
		return ;
	}
	push_down(p,pl,pr);
	int mid=(pl+pr)>>1;
	if(mid>=L)updata(ls(p),L,R,pl,mid);
	if(mid<R)updata(rs(p),L,R,mid+1,pr);
	push_up(p);
	return ;
}
int query(int p,int L,int R,int pl,int pr){
	if(pl>=L&&pr<=R)return tre[p];
	push_down(p,pl,pr);
	int res=0;
	int mid=(pl+pr)>>1;
	if(mid>=L)res+=query(ls(p),L,R,pl,mid);
	if(mid<R)res+=query(rs(p),L,R,mid+1,pr);
	return res;
}
signed main(){
	scanf("%d%d",&n,&m);
	while(m--){
		int op,l,r;
		scanf("%d%d%d",&op,&l,&r);
		if(op==0){
			updata(1,l,r,1,n);
		}
		else{
			printf("%d\n",query(1,l,r,1,n));
		}
	}
	return 0;
}
2023/11/4 16:28
加载中...