Codeforces149E - Martian Strings(KMP)

题目大意

给定一个字符串T，接下来有n个字符串，对于每个字符串S,判断是否存在T[a…b]+T[c…d]=S(1 ≤ a ≤ b < c ≤ d ≤ length(T))

题解

对于每个字符串S,我们先顺序在字符串T上进行匹配，并记录S的每个字符第一次在T上匹配到的位置（假设用数组pos记录），然后把字符串T和S都逆序，再用S在T上进行匹配，假设当前已经S匹配到第i个，在字符串T上的位置为j,S的第i+1个字符在原字符串中的位置为length(S)-i-2,T字符串的第j个字符在原字符串中的位置为length(T)-j-1,如果pos[length(S)-i-2]<length(T)-j-1，则符合条件

代码：

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

using namespace std;

#define MAXN 1005

char p[MAXN],T[MAXN*100];

int f[MAXN],pos[MAXN],ans;

void getfail()

{

    int j,len=strlen(p);

    f[0]=j=-1;

    for(int i=1;i<len;i++)

    {

        while(j>=0&&p[j+1]!=p[i]) j=f[j];

        if(p[j+1]==p[i]) j++;

        f[i]=j;

    }

}

void find(int d)

{

    int j=-1,lent=strlen(T),lenp=strlen(p);

    for(int i=0;i<lent;i++)

    {

        while(j>=0&&p[j+1]!=T[i]) j=f[j];

        if(p[j+1]==T[i]) 

        {

            j++;

            if(!d&&pos[j]==-1) pos[j]=i;

            if(d)

            {

                if(lenp-j-2>=0&&pos[lenp-j-2]!=-1)

                {

                    if(pos[lenp-j-2]<lent-i-1)

                    {

                        ans++;

                        break;

                    }

                }

            }

        }

        if(!d&&j+1==lenp) break;

    }

}

int main()

{

    int n;

    scanf("%s",T);

    scanf("%d",&n);

    ans=0;

    for(int i=0;i<n;i++)

    {

        memset(pos,-1,sizeof(pos));

        scanf("%s",p);

        getfail();

        find(0);

        reverse(p,p+strlen(p));

        reverse(T,T+strlen(T));

        getfail();

        find(1);

        reverse(T,T+strlen(T));

    }

    printf("%d\n",ans);

    return 0;

}