【codevs1690】開關燈
阿新 • • 發佈:2018-10-11
ons bit -- printf sin max print build else if
這道題需要支持區間修改和區間詢問,因此采用線段樹加以維護。
由於求的是開著的燈的數目,因此維護sum:區間[ l , r ]中開著的燈的數目。
tag取做0/1,表示區間是否反轉,在進行標記下傳時,如果tag=1,則下傳,否則返回。(tag的選取一般跟操作有關)
代碼如下:
#include <bits/stdc++.h> using namespace std; const int maxn=1e5+10; int n,m; struct node{ #define lson t[k].lc #define rson t[k].rc int lc,rc,sum,tag; }t[maxn<<1]; int tot=1; inline void pushup(int k){t[k].sum=t[lson].sum+t[rson].sum;} inline void pushdown(int k,int l,int r){ int mid=l+r>>1; if(t[k].tag==0)return; t[lson].sum=mid-l+1-t[lson].sum,t[lson].tag^=1; t[rson].sum=r-mid-t[rson].sum,t[rson].tag^=1; t[k].tag=0; } void build(int k,int l,int r){ if(l==r)return; int mid=l+r>>1; lson=++tot,build(lson,l,mid); rson=++tot,build(rson,mid+1,r); } void modify(int k,int l,int r,int x,int y){ if(l==x&&r==y){t[k].sum=r-l+1-t[k].sum,t[k].tag^=1;return;} int mid=l+r>>1; pushdown(k,l,r); if(y<=mid)modify(lson,l,mid,x,y); else if(x>mid)modify(rson,mid+1,r,x,y); else modify(lson,l,mid,x,mid),modify(rson,mid+1,r,mid+1,y); pushup(k); } int query(int k,int l,int r,int x,int y){ if(l==x&&r==y)return t[k].sum; int mid=l+r>>1; pushdown(k,l,r); if(y<=mid)return query(lson,l,mid,x,y); else if(x>mid)return query(rson,mid+1,r,x,y); else return query(lson,l,mid,x,mid)+query(rson,mid+1,r,mid+1,y); } void read_and_parse(){ scanf("%d%d",&n,&m); build(1,1,n); } void solve(){ int opt,x,y; while(m--){ scanf("%d%d%d",&opt,&x,&y); if(opt==0)modify(1,1,n,x,y); else printf("%d\n",query(1,1,n,x,y)); } } int main(){ read_and_parse(); solve(); return 0; }
【codevs1690】開關燈