HdU 4046 Panda 段树

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4046

意甲冠军：到了bw组成的长度为n的字符串（n<=50000）。有m次操作（m<=10000），每次操作是询问一段范围内wbw的个数。或者改变一个字符成为w或b。

思路：建一棵线段树，每一个结点记录的是从L到R以每一个i为最左边的字母的总共的wbw的个数，单点更新的时候要更新三个点。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <cstdlib>
#include <queue>
#include <stack>
#include <vector>
#include <ctype.h>
#include <algorithm>
#include <string>
#include <set>
#include <ctime>
#define PI acos(-1.0)
#define INF 0x7fffffff
#define eps 1e-8
#define maxn 50005
typedef long long LL;
typedef unsigned long long ULL;
using namespace std;
int n,m,l,r;
int num[maxn*4];
char ss[maxn*4];
struct line
{
    int left,right;
    int n;
    char value;
} a [maxn*4];
void build_tree(int l,int r,int step)
{
    a[step].left=l;
    a[step].right=r;
    a[step].n=0;
    if(l==r)
    {
        a[step].n=((l+2<=n)&&(ss[l]=='w'&&ss[l+1]=='b'&&ss[l+2]=='w'));
        num[l]=step;
        return;
    }
    int mid=(l+r)>>1;
    build_tree(l,mid,step<<1);
    build_tree(mid+1,r,step<<1|1);
    a[step].n=a[step<<1].n+a[step<<1|1].n;
}
int ans;
void change(int s,int step,char k)
{
    if(s==a[step].left&&s==a[step].right)
    {
        a[step].value=k;
        ss[s]=k;
        a[step].n=((l+2<=n)&&ss[s]=='w'&&ss[s+1]=='b'&&ss[s+2]=='w');
        return;
    }
    if(a[step].left==a[step].right) return;
    int mid=(a[step].left+a[step].right)>>1;
    if(mid>=s) change(s,step<<1,k);
    else change(s,step<<1|1,k);
    a[step].n=a[step<<1].n+a[step<<1|1].n;
}
void change2(int s,int step)
{
    if(s==a[step].left&&s==a[step].right)
    {
        a[step].n=(ss[s]=='w'&&ss[s+1]=='b'&&ss[s+2]=='w');
        return;
    }
    if(a[step].left==a[step].right) return;
    int mid=(a[step].left+a[step].right)>>1;
    if(mid>=s) change2(s,step<<1);
    else change2(s,step<<1|1);
    a[step].n=a[step<<1].n+a[step<<1|1].n;
}
int query(int s,int t,int step)
{
    if(a[step].left>=s&&a[step].right<=t)
    {
        ans+=a[step].n;
        return 0;
    }
    else
    {
        int mid=(a[step].left+a[step].right)/2;
        if(mid>=t) query(s,t,step<<1);
        else if(mid<s) query(s,t,step<<1|1);
        else
        {
            query(s,t,step<<1);
            query(s,t,step<<1|1);
        }
    }
}
int main()
{
    int T,q,L,R,p;
    char k[2];
    scanf("%d",&T);
    for(int ii=1;ii<=T;ii++)
    {
        scanf("%d%d",&n,&m);
        scanf("%s",ss+1);
        printf("Case %d:
",ii);
        build_tree(1,n,1);
        for(int i=0;i<m;i++)
        {
            scanf("%d",&q);
            if(q==0)
            {
                ans=0;
                scanf("%d%d",&L,&R);
                if(R-1<L+1)
                {
                    printf("0
");
                    continue;
                }
                query(L+1,R-1,1);
                printf("%d
",ans);
            }
            else
            {
                scanf("%d%s",&p,&k);
                change(p+1,1,k[0]);
                if(p>=1)
                change2(p,1);
                if(p>=2)
                change2(p-1,1);
            }
        }
    }
    return 0;
}

版权声明：本文博客原创文章。博客，未经同意，不得转载。