「CQOI2006」简单题 线段树

「CQOI2006」简单题 线段树

水。区间修改,单点查询。用线段树维护区间\([L,R]\)内的所有\(1\)的个数,懒标记表示为当前区间是否需要反转(相对于区间当前状态),下方标记时懒标记取反即可。

#include <cstdio>
#include <cmath>
#define sl (x<<1)
#define sr (x<<1|1)
#define MAXN 100010
using namespace std;
struct nod{
int l; int r;
bool lazy; int val;
} tre[MAXN*4];
void buildt(int l, int r, int x){
tre[x].l=l,tre[x].r=r;
if(l==r){
tre[x].val=0;
return;
}
int mid=(l+r)>>1;
buildt(l, mid, x<<1);
buildt(mid+1, r, x<<1|1);
}
void push_down(int x){
if(tre[x].lazy==0) return;
tre[sl].lazy=!tre[sl].lazy;
tre[sr].lazy=!tre[sr].lazy;
if(tre[x].lazy) tre[sl].val=(tre[sl].r-tre[sl].l+1)-tre[sl].val;
if(tre[x].lazy) tre[sr].val=(tre[sr].r-tre[sr].l+1)-tre[sr].val;
tre[x].lazy=0;
}
void change(int x, int ql, int qr){
if(ql<=tre[x].l&&tre[x].r<=qr){
tre[x].val=tre[x].r-tre[x].l+1-tre[x].val;
tre[x].lazy=!tre[x].lazy;
return;
}
push_down(x);
int mid=(tre[x].l+tre[x].r)>>1;
if(ql<=mid) change(sl,ql,qr);
if(qr>mid) change(sr,ql,qr);
tre[x].val=tre[sl].val+tre[sr].val;
}
int query(int x, int q){
if(tre[x].l==tre[x].r){
return tre[x].val;
}
push_down(x);
int mid=(tre[x].l+tre[x].r)>>1;
int ans=0;
if(q<=mid) ans=query(sl,q);
else ans=query(sr,q);
tre[x].val=tre[sl].val+tre[sr].val;
return ans;
}
int n,m;
int main()
{
scanf("%d %d", &n, &m);
buildt(1,n,1);
while(m--){
int t;
scanf("%d", &t);
if(t==1){
int l,r;
scanf("%d %d", &l, &r);
change(1,l,r);
}else{
int i;scanf("%d", &i);
printf("%d\n", query(1,i));
}
}
return 0;
}
上一篇:LOJ 2980 「THUSCH 2017」大魔法师——线段树


下一篇:windows10安装anaconda,配置tensorflow