洛谷 P2574 XOR的藝術
阿新 • • 發佈:2018-12-17
思路: 線段樹。
注意: 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;
}