Codeforces 149E Martian Strings-字符串匹配

传送门

题意：

给出一个长度为n的文本串和m个模式串，求有多少个模式串可以拆成两半，使得这两半按顺序匹配

(n<=2e5,m<=100)

Solution：

先正着跑一遍KMP,pos[i]表示模式串中最早能匹配到第i个字符的位置

然后把文本串和模式串都取反，再跑一遍KMP，求出文本串的每个位置i所能匹配的最长的模式串长度z[i]，设模式串长为len,那么我们只需要比较i和pos[len-z[i]]的关系即可，具体推一推即可

注意特判len=1的情况

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int z[100010],n,r,l;
int maxn[200010],pos[200010];
char b[200010],a[100010];
int ans=0;
int main()
{
    scanf("%s",a);
    int m=strlen(a);
    scanf("%d",&n);
    for (int T=1;T<=n;T++)
    {
        scanf("%s",b);
        int len=strlen(b);
        if (len<2) continue;
        for (int i=0;i<m;i++)
            b[1+i+len]=a[i];
        l=r=0;
        for (int i=1;i<=m+len;i++)
        {
            if (i<=len)pos[i]=1e9;
            if (i>r)
            {
                int t=i,bg=0;
                while (b[t]==b[bg])
                    t++,bg++;
                z[i]=bg;
                r=t-1;
                l=i;
            }
            else
            {
                if (z[i-l]<r-i+1) z[i]=z[i-l];
                else
                {
                    int t=r+1,bg=r-i+1;
                    while (b[t]==b[bg])
                        t++,bg++;
                    z[i]=bg;
                    r=t-1;
                    l=i;
                }
            }
            if (z[i]&&i>len) pos[z[i]]=min(pos[z[i]],i-len+z[i]-1);
        }
        for (int i=len-1;i>=1;i--)
            pos[i]=min(pos[i+1]-1,pos[i]);
        for (int i=0,j=len-1;i<j;i++,j--)
            swap(b[i],b[j]);
        b[len]='$';
        for (int i=len+1,j=m+len;i<j;i++,j--)
            swap(b[i],b[j]);
        l=r=0;
        for (int i=1;i<=m+len;i++)
        {
            if (i>r)
            {
                int t=i,bg=0;
                while (b[t]==b[bg])
                    t++,bg++;
                z[i]=bg;
                r=t-1;
                l=i;
            }
            else
            {
                if (z[i-l]<r-i+1) z[i]=z[i-l];
                else
                {
                    int t=r+1,bg=r-i+1;
                    while (b[t]==b[bg])
                        t++,bg++;
                    z[i]=bg;
                    r=t-1;
                    l=i;
                }
            }
            if (i>len)if (pos[len-z[i]]<=m+len-i-z[i]+1) {ans++;break;}
        }       
    }
    printf("%d",ans);
}