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;
}