hdu 4046 Panda 线段树

五场的网络赛的题目出现了大量的线段树的题目，每次都只能坐在那里干瞪眼。。

看来需要好好的啃啃线段树了。。

 

#include<iostream>
#include<string> 
#define maxn 50010 
#define lson l , m , rt << 1 
#define rson m + 1 , r , rt << 1 | 1 
using namespace std; 
int sum[maxn*4]; 
char str[maxn]; 
int a[maxn]; 
void PushUP(int rt)
 { 
    sum[rt] = sum[rt<<1] + sum[rt<<1|1]; 
} 
void build(int l,int r,int rt)
 { 
    if (l == r) { 
        sum[rt]=a[l]; 
        return ; 
    } 
    int m = (l + r) >> 1; 
    build(lson); 
    build(rson); 
    PushUP(rt); 
} 

void update(int p,int add,int l,int r,int rt)
 { 
    if (l == r)
 { 
        sum[rt] = add; 
        return ; 
    } 
    int m = (l + r) >> 1; 
    if (p <= m) update(p , add , lson); 
    else update(p , add , rson); 
    PushUP(rt); 
} 
int query(int L,int R,int l,int r,int rt)  
{ 
    if (L <= l && r <= R) { 
        return sum[rt]; 
    } 
    int m = (l + r) >> 1; 
    int ret = 0; 
    if (L <= m) ret += query(L , R , lson); 
    if (R > m) ret += query(L , R , rson); 
    return ret; 
} 
int main() 
{ 
    int cas,T=0,n,m,r,l,c; 
    char st; 
    scanf("%d",&cas); 
    while(cas--) 
    { 
        scanf("%d %d",&n,&m); 
        scanf("%s",str); 
        memset(a,0,sizeof(a)); 
        printf("Case %d:\n",++T); 
        for(int i=2;i<n;i++)//标记以该下标结尾的长度为3的子串是否符合要求 
        { 
            if(str[i]!='w') 
                a[i]=0; 
            else if(str[i-1]=='b'&&str[i-2]=='w') 
                a[i]=1; 
        } 
        build(0,n,1); 
        while(m--) 
        { 
            scanf("%d",&c); 
            if(c==0) 
            { 
                scanf("%d %d",&l,&r); 
    l+=2; 
                if(l>r) printf("0\n"); 
                else 
     printf("%d\n",query(l,r,0,n,1)); 
            } 
            else 
            { 
                scanf("%d %c",&l,&st); 
                if(str[l]==st)//若要就该的字符跟原先相同，则不用修改了 
                    continue; 
                str[l]=st; 
                if(l>=2&&str[l-2]=='w'&&str[l-1]=='b'&&str[l]=='w') 
                { 
                    if(a[l]==0)//若改之前是不符合要求的串，则更新 
                        update(l,1,0,n,1); 
                    a[l]=1; 
                } 
                else if(l>=2&&a[l]==1)//若改之前是符合要求的串，则更新 
                { 
                    update(l,0,0,n,1); 
                    a[l]=0; 
                } 
                if(l>=1&&str[l]=='b'&&str[l-1]=='w'&&str[l+1]=='w') 
                { 
                    if(a[l+1]==0) 
                        update(l+1,1,0,n,1); 
                    a[l+1]=1; 
                } 
                else if(l<n-1&&l>=1&&a[l+1]==1) a[l+1]=0,update(l+1,0,0,n,1); 
                if(str[l]=='w'&&str[l+1]=='b'&&str[l+2]=='w') 
                { 
                    if(a[l+2]==0) 
                        update(l+2,1,0,n,1); 
                    a[l+2]=1; 
                } 
                else
     if(l<n-2&&a[l+2]==1)
      a[l+2]=0,update(l+2,0,0,n,1); 
            } 
        } 
    } 
    return 0; 
}