【codevs1690】開關燈

標籤：parse   nod   print   int   tag   cpp   clu   oid   pre   

這道題需要支援區間修改和區間詢問，因此採用線段樹加以維護。
由於求的是開著的燈的數目，因此維護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】開關燈