2017 ACM-ICPC 亚洲区（乌鲁木齐赛区）网络赛-G: Query on a string(线段树+预处理)

题目链接：暂无

题面：

题目大意：

给你一个A串一个B串，然后进行两种操作。

（1）：给出p，q查询在A串中 p，q之间有多少B串。

（2）：给出 p ch，将p位置的字符更改为ch。

解题思路：

刚开始比赛过去两个小时都没什么思路，后来还是队友给我讲了他的想法我才写出来，还是弱啊Orz，

其实就是把A串先预处理成一个01串，01就表示在A串中从当前字符开始是否是B串，是的话就是1，有点讲不清楚，举个例子

AAABBCC

AA

那么A串就可以转化为1100000，这样预处理之后就可以直接向上维护一个求和的线段树，查询也直接查询即可。

更新操作的话需要注意一点，不仅仅更新当前点，还要考虑这个点之前的01可能会变化的点，更新即可，整体过程用线段树维护01串。

以下贴代码，因为是比赛时候写的，所以巨丑无比。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
#include <set>
#include <functional>
#define rank ra
#define pb push_back
#define lson rt<<1
#define rson rt<<1|1
using namespace std;
typedef long long ll;
const int N=100010;
const int INF=1e9+7;
char s[100005],g[15];   //储存两个串
int s1[100005];     //储存01串
char q1,ch;
int m,p,q,lg,ans,ls;
struct node
{
    int l,r,mid;
    int cnt;
}t[N<<2];
void pushup(int rt)     //正常的线段树pushup
{
    t[rt].cnt=t[lson].cnt+t[rson].cnt;
}
void build(int l,int r,int rt)
{
    int m=(l+r)>>1;
    t[rt].l=l;t[rt].r=r;
    t[rt].mid=m;
    if(l==r)
    {
        int flag=1;
        for(int i=l,j=0;j<lg;i++,j++)
        {
            if(s[i]!=g[j])
            {
                flag=0;
                break;
            }
        }
        if(flag)
            t[rt].cnt=1;
        else
            t[rt].cnt=0;
        s1[l]=t[rt].cnt;    //建树的时候直接更新s1数组
        return ;
    }
    build(l,m,lson);
    build(m+1,r,rson);
    pushup(rt);
}
void update(int pos,int flag,int rt)    //正常的线段树更新
{
    if(t[rt].l==t[rt].r)
    {
        t[rt].cnt=flag;
        return ;
    }
    if(pos<=t[rt].mid)
        update(pos,flag,lson);
    if(pos>t[rt].mid)
        update(pos,flag,rson);
    pushup(rt);
}
void query(int p,int q,int rt)  //正常的线段树查询
{
    if(p<=t[rt].l&&t[rt].r<=q)
    {
        ans+=t[rt].cnt;
        return ;
    }
    if(p<=t[rt].mid)
        query(p,q,lson);
    if(q>t[rt].mid)
        query(p,q,rson);
}
int main()
{
    int QAQ;
    scanf("%d",&QAQ);
    while(QAQ--)
    {
        scanf("%d",&m);
        scanf(" %s %s",s,g);
        ls=strlen(s);
        lg=strlen(g);
        build(0,ls,1);
        while(m--)
        {
            scanf(" %c",&q1);
            if(q1=='Q')
            {
                scanf("%d%d",&p,&q);
                ans=0;
                q=q-lg+1;   //这里注意因为缩点所以q的范围要变化
                if(p<=q)
                query(p-1,q-1,1);
                printf("%d\n",ans);
            }
            int lv;
            if(q1=='C')
            {
                scanf("%d %c",&p,&ch);  //更新操作
                s[p-1]=ch;
                lv=p-lg;
                lv=max(lv,0);
                for(int i=lv;i<=p;i++)  //枚举所有可能发生变化的点
                {
                    int flag=1;
                    for(int j=0,k=i;j<lg;j++,k++)
                    {
                        if(s[k]!=g[j])
                        {
                            flag=0;
                            break;
                        }
                    }
                    if((flag&&s1[i]==0)||(flag==0&&s1[i]==1))   //这里小优化，只有跟原先01不一样才更新
                    {
                        update(i,flag,1);
                        s1[i]=flag;
                    }
                }
            }
        }
        printf("\n");
    }
}