【题目链接】
【题目大意】
给定一个长度为n(n<=10^5)个01序列,定义以下两种操作:
①:将区间[l,r]的序号01翻转,0->1,1->0,如101翻转后为010
②:求区间[l,r]中满足l <= i < j <= r 且 the ith element is 0, jth element is 1
操作次数不超过10^5次。
【分析】
此题维护一个区间上的某些操作,使用线段树比较适合。
对于线段树,通常要维护边界[l,r]。
对于操作②,要求出[l,r]满足其条件的,取mid=(l+r)>>1,query[l,r] = query[l,mid]+query[mid+1,r]+[l,mid]中0的个数*[mid+1]中1的个数。
于是还需要维护一个域存储本区间0的个数,同时存储本区间的query值。
对于操作①,要求出反转区间之后区间域的变化,目测翻转后的值和之前的值并没有太大关系,故增加一个域存储反转后的值,这样反转的时候只需要互换这两个值即可。该域值定义如下:
区间[l,r]中满足l <= i < j <= r 且 the ith element is1, jth element is 0。
同时为了实现惰性操作,还需要一个bool变量rev来标记本区间是否需要反转。
于是维护如下域:
int l , r ; //边界
int zeroCnt ; //[l,r]中0的数目
unsigned int value ; //query的值
unsigned int anti_value ; //反转后的query值
bool rev ; //标记是否需要翻转
【代码】
//2012.11.18 线段树 soj karsbin #include <stdio.h> #include <ctype.h> #include <string.h> const int maxn = 1024 * 100 ; inline bool get(int &t) { bool flag = 0 ; char c; while(!isdigit(c = getchar())&&c!='-') if( c == -1 ) break ; if( c == -1 ) return 0 ; if(c=='-') flag = 1 , t = 0 ; else t = c ^ 48; while(isdigit(c = getchar())) t = (t << 1) + (t << 3) + (c ^ 48) ; if(flag) t = -t ; return 1 ; } struct node { int l , r ; //边界 int zeroCnt ; //[l,r]中0的数目 unsigned int value ; //query的值 unsigned int anti_value ; //反转后的query值 bool rev ; //标记是否需要翻转 }; node tree[maxn*4] ; int top , n , m , len ; char str[maxn] ; void change(int root) { if ( tree[root].l < tree[root].r ) { int mid = tree[root*2].r ; tree[root].zeroCnt = tree[root*2].zeroCnt + tree[root*2+1].zeroCnt ; tree[root].value = tree[root*2].value + tree[root*2+1].value + tree[root*2].zeroCnt * (tree[root*2+1].r-mid-tree[root*2+1].zeroCnt) ; tree[root].anti_value = tree[root*2].anti_value + tree[root*2+1].anti_value + (mid-tree[root*2].l+1-tree[root*2].zeroCnt)*tree[root*2+1].zeroCnt ; } } void build(int root,int l,int r) { tree[root].l = l ; tree[root].r = r ; if ( l == r ) { tree[root].zeroCnt = str[l-1] ^ 49 ; tree[root].value = tree[root].anti_value = 0 ; } else { int mid = ( l + r ) >> 1 ; build(root*2, l, mid); build(root*2+1, mid+1, r); change(root); } tree[root].rev = false ; } void modify(int root) { if ( tree[root].rev == true ) { tree[root].zeroCnt = tree[root].r - tree[root].l + 1 - tree[root].zeroCnt ; unsigned int t = tree[root].anti_value ; tree[root].anti_value = tree[root].value ; tree[root].value = t ; if ( tree[root].l < tree[root].r ) { tree[root*2].rev ^= 1 ; tree[root*2+1].rev ^= 1 ; } tree[root].rev = false ; } } unsigned int query(int root,int l,int r,int& zero, int &one) { modify(root); if ( tree[root].l == l && tree[root].r == r ) { zero = tree[root].zeroCnt ; one = tree[root].r - tree[root].l + 1 - zero ; return tree[root].value ; } int mid = ( tree[root].l + tree[root].r ) >> 1 ; unsigned int ans = 0; if( r <= mid ) ans = query(root*2, l, r, zero, one); else if( l > mid ) ans = query(root*2+1, l, r, zero, one); else { int z1, z2 ; int o1, o2 ; ans = query(root*2, l, mid, z1, o1)+query(root*2+1, mid+1, r, z2, o2); zero = z1 + z2 ; one = o1 + o2 ; ans += z1 * o2 ; } modify(root*2); modify(root*2+1); change(root); return ans ; } void reverseSeq(int root,int l,int r) { modify(root); if ( tree[root].l == l && tree[root].r == r ) { tree[root].rev ^= 1 ; modify(root); return ; } int mid = ( tree[root].l + tree[root].r ) >> 1 ; if( r <= mid ) reverseSeq(root*2, l, r); else if( mid < l ) reverseSeq(root*2+1, l, r); else { reverseSeq(root*2, l, mid); reverseSeq(root*2+1, mid+1, r); } modify(root*2); modify(root*2+1); change(root); } int main() { int t , op , l , r ; get(t); while (t--) { get(n); get(m); scanf("%s",str); build(1,1,strlen(str)); while (m--) { get(op); get(l); get(r); l++;r++; if ( op == 0 ) { reverseSeq(1,l,r); } else { int zero , one ; printf("%u\n",query(1,l,r,zero,one)); } } } }