1. 程式人生 > >洛谷 P2574 XOR的藝術

洛谷 P2574 XOR的藝術

思路: 線段樹。

注意: 1、所有和線段樹相關的陣列開十倍。 2、push_down之後lazy標記要清空。

程式碼:

#include<bits/stdc++.h>
using namespace std;

#define maxn 200000
#define read(x) scanf("%d",&x)

#define lch (o*2)
#define rch (o*2+1)
#define mid (l+(r-l)/2)

int n,m;
int c[maxn+5];
int a[maxn*10],lzy[maxn*10],sz[maxn*10];

int P,Q;

void
push_up(int o) { a[o]=a[lch]+a[rch]; sz[o]=sz[lch]+sz[rch]; } void push_down(int o) { if(lzy[o]) { lzy[o]=0; a[lch]=sz[lch]-a[lch],a[rch]=sz[rch]-a[rch]; lzy[lch]^=1,lzy[rch]^=1; } } void make_tree(int o,int l,int r) { if(l==r) { a[o]=c[l]; sz[o]=1; return ; } make_tree(lch,l,mid)
,make_tree(rch,mid+1,r); push_up(o); } void update(int o,int l,int r) { push_down(o); if(l>=P&&r<=Q) { lzy[o]^=1; a[o]=sz[o]-a[o]; return ; } if(Q<l||P>r) return ; update(lch,l,mid),update(rch,mid+1,r); push_up(o); } int query(int o,int l,int r) { push_down(o); if
(l>=P&&r<=Q) return a[o]; if(Q<l||P>r) return 0; int x=query(lch,l,mid)+query(rch,mid+1,r); return x; } int main() { read(n),read(m); for(int i=1; i<=n; i++) { char x; while(-1!=scanf("%c",&x)&&x!='0'&&x!='1'); c[i]=x-'0'; } make_tree(1,1,n); while(m--) { int opr; read(opr),read(P),read(Q); if(opr==0) { update(1,1,n); } else { int x=query(1,1,n); printf("%d\n",x); } } return 0; }