17年ICPC新疆网络赛G Query on a string(线段树 KMP)

• 题目大意

  给一个字符串S(长度最大100000)和T(最长10)
  给出n次操作，一共有两种操作:

  1. Q i j :计算S中区间[i,j]中有多少个子区间能和T匹配
  2. C x ch :将S[x]变成字符ch

• 分析
  我们用A数组记录下S中以每个下标为开头的字符串是否匹配，1表示匹配，0表示不匹配
  这样就可以通过线段树来查询一个区间中有多少匹配的子区间了(维护一个区间最大值)
  然后就是更新的操作了，因为T最长是10,所以我们知道在改变S中一个点的时候最多会影响到10个区间，通过KMP处理一下被影响的区间的A数组的值即可。

• 代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=100007;
typedef long long int LL;
int n;
char S[MAXN];
char T[15];
int S_len;
int T_len;
int Next[15];
int A[MAXN];//下标从1开始
int sum[MAXN*4];
void Pushup(int rt)//Pushup函数假设rt的儿子都已经更新好了，对rt进行更新
{
    sum[rt]=sum[rt*2]+sum[rt*2+1];
}
void Build(int l,int r,int rt)//建树
{
    if(l==r){sum[rt]=A[l];return ;}

    int m=(l+r)/2;
    Build(l,m,rt*2);
    Build(m+1,r,rt*2+1);
    Pushup(rt);
}
void Update(int X,int C,int l,int r,int rt)//将某个节点X修改为C
{
    if(l==r){sum[rt]=C;return ;}

    int m=(l+r)/2;
    if(X<=m)Update(X,C,l,m,rt*2);
    else Update(X,C,m+1,r,rt*2+1);
    Pushup(rt);
}
int Query(int L,int R,int l,int r,int rt)//L R为查询区间,该函数返回(L,R)与(l,r)交集中的最大值
{
    int ans=0;
    int m=(l+r)/2;
    if(L<=l && r<=R)return sum[rt];
    if(L<=m)ans+=Query(L,R,l,m,rt*2);
    if(R>m)ans+=Query(L,R,m+1,r,rt*2+1);
    return ans;

}

void Init()
{
    memset(A,0,sizeof(A));
}
void In()
{
    scanf("%d",&n);
    scanf("%s",S);
    scanf("%s",T);
    S_len=strlen(S);
    T_len=strlen(T);
}
void Make_Next(char P[],int Next[])
{
    int i,j;
    j=Next[0]=-1;i=0;
    while (i<strlen(P))
    {
        while (j!=-1&&P[i]!=P[j]) j=Next[j];
        Next[++i]=++j;
    }
}
void get_next(char *t,int *next){
    int len =strlen(t);
    int i=0,j=-1;
    next[0]=-1;
    while(i<len){
        if(j==-1 || t[i]==t[j]) i++,j++,next[i]=j;
        else j=next[j];
    }
}
void KMP(char *s,char *t,int *next,int from,int to,int *ans){
    for(int i=from;i<=to;i++){
        if(ans[i+1]==1)
            {
                ans[i+1]=0;
                Update(i+1,0,1,S_len,1);
        }
    }
    int i=from,j=0,len=strlen(t);
    while(i<to+len){
        if(j==-1 || s[i]==t[j]){
            if(j==len-1){
                ans[i-j+1]=1;
                Update(i-j+1,1,1,S_len,1);
                j=next[j];
            }
            else i++,j++;
        }
        else j=next[j];
    }
}
void Work()
{
    char c;
    int x,y;
    for(int i=1;i<=n;i++)
    {
        scanf("%c",&c);

        while(c=='\n' || c==' ')scanf("%c",&c);
        if(c=='Q')
        {
            scanf("%d%d",&x,&y);
            if(y-T_len+1<x)printf("0\n");
            else printf("%d\n",Query(x,y-T_len+1,1,S_len,1));
        }
        else
        {
            scanf("%d",&x);
            scanf("%c",&c);
            while(c=='\n' || c==' ')scanf("%c",&c);
            S[x-1]=c;
            int from=max(x-T_len+1-1,0);
            int to=min(x,S_len-T_len+1-1);
            KMP(S,T,Next,from,to,A);
        }
    }
}
void Match()
{

    for(int i=0;i<=S_len-T_len;i++)
    {
        for(int j=0;j<T_len;j++)
        {
            if(S[i+j]!=T[j]){A[i+1]=0;break;}
            if(j==T_len-1)A[i+1]=1;
        }
    }
}
int main()
{
    int Test;
    scanf("%d",&Test);
    while(Test--)
    {
        Init();
        In();
        get_next(T,Next);
        Match();
        Build(1,S_len,1);
        Work();
        if(Test!=0)printf("\n");
    }
    return 0;
}