关闭

【JZOJ 3427】归途与征程

标签: kmp字符串
530人阅读 评论(0) 收藏 举报
分类:

Description

这里写图片描述

Solution

题意简化以后就是:在b串的同构中,有多少个可以按顺序匹配多个串。
我们可以先把a串拆开,每个做一遍KMP的预处理,
倍长b串,枚举b串的开头,依次匹配过去,
每次匹配的时候,记录一下上一次匹配失败的位置,直接从上次匹配失败的地方开始匹配;成功是也记录一下,判断一下当前可不可以用那个匹配成功的去匹配当前的位置。
复杂的:O(nm);

Code

#include<cstdio>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=100500;
int m,n,ans;
char b[N*2],a[N];
int nx[N],A[N][2],A0;
int B[N],L[N];
int max(int a,int b){return a<b?b:a;}
bool ss(int q,int l,int r)
{
    l=max(l,L[q]);
    if(A0==1&&l!=r+1)return 0;
    if(q>A0 && l<=r+1)return 1;
    if(r-l<A[q][1]-A[q][0])return 0;
    if(q==A0 && a[n]!='*')
    {
        int w=r-(A[q][1]-A[q][0])-A[q][0];
        fo(i,A[q][0],A[q][1])if(a[i]!=b[w+i])return 0;
        return 1;
    }
    int j=A[q][0]-1;
    if(B[q]>=l)return ss(q+1,B[q]+A[q][1]-A[q][0],r);
    fo(i,l,r)
    {
        while(j>=A[q][0]&&a[j+1]!=b[i])j=nx[j];
        if(a[j+1]==b[i])j++;
        if(j>=A[q][1])
        {
            bool z=ss(q+1,i+1,r);
            B[q]=z*(i-j+2);
            return z;
        }
    }
    L[q]=max(L[q],r-j);
    return 0;
}
int main()
{
    int q,w;char ch;
    for(ch=getchar();ch<='z'&&ch>='a'||ch=='*';ch=getchar())a[++n]=ch;
    a[n+1]=a[0]='*';
    for(;!(ch<='z'&&ch>='a');ch=getchar());
    for(;ch<='z'&&ch>='a';ch=getchar())b[++m]=ch;
    fo(i,1,m)b[m+i]=b[i];
    fo(i,1,n)if(a[i]!='*')
    {
        if(a[i-1]=='*')A[++A0][0]=i;
        A[A0][1]=i;
    }
    fo(I,1,A0)
    {
        int j=A[I][0]-1;
        nx[A[I][0]]=j;
        fo(i,A[I][0]+1,A[I][1])
        {
            while(j>=A[I][0]&&a[j+1]!=a[i])j=nx[j];
            if(a[j+1]==a[i])j++;
            nx[i]=j;
        }
    }
    ans=0;
    if(!A0)ans=m;else
    for(int i=1,j=0;i-j<=m;i++)
    if(a[1]=='*')ans+=ss(1,i,i+m-1);else
    {
        while(j>=A[1][0]&&a[j+1]!=b[i])j=nx[j];
        if(a[j+1]==b[i])j++;
        if(j>=A[1][1])
        {
            ans+=ss(2,i+1,i-j+m);
            j=nx[j];
        }
    }
    printf("%d\n",ans);
    return 0;
}
1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:71666次
    • 积分:3325
    • 等级:
    • 排名:第10192名
    • 原创:260篇
    • 转载:2篇
    • 译文:0篇
    • 评论:39条
    博客公告
    博主的智商比较低,人比较蠢,因此神犇们如发现任何的错误欢迎拍打喂食吐槽
    --------
    【JZOJ 2700】【GDKOI2012模拟02.01】数字(中国剩余定理)
    博主状态

    Too Weak,
    so
    Fighting!!!

    链接
    最新评论