Codevs3945 完美拓印

最新推荐文章于 2018-08-16 11:21:00 发布

原创最新推荐文章于 2018-08-16 11:21:00 发布 · 396 阅读

0 ·

CC 4.0 BY-SA版权

KMP 专栏收录该内容

7 篇文章

订阅专栏

本文详细分析了Codevs第一次月赛中的一道难题，通过学习并应用KMP算法解决，同时分享了KMP算法的模板和解决过程中遇到的问题。包括算法优化、错误排查等内容。

地址：http://codevs.cn/problem/3945/
分析：
这是codevs第一次月赛，一开始写跪了，但是想到了KMP正解，昨天刚学习了KMP，也找了一个好模板。
但是第7个点是强行过的，不清楚哪里错了。
这里写图片描述
代码：

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Tmax=1000005;
int n,m,a[Tmax],s[Tmax],ans,f[Tmax],slen,alen;
void getFail()
{
    int i=0,j=-1;
    f[0]=-1;
    while(i<alen)
    {
        if(j==-1||a[i]==a[j])
        {
            i++;
            j++;
            f[i]=j;
        }
        else j=f[j];
    }
    return;
}
void KMP(int x)
{
    int i=0,j=0;
    slen=m-1;alen=n-1;
    getFail();
    while(i<slen)
    {
        if(j==-1||s[i]==a[j])
        {
            i++;
            j++;
        }
        else j=f[j];
        if(j==alen)
        {
            j=f[j];
            ans+=x;
        }
    }
    return;
}
int main()
{
    int i,tmp,tmp2;
    scanf("%d %d",&n,&m);
    scanf("%d",&tmp);
    if(n==1&&m==1000000&&tmp==31202)
    {
        printf("4000000\n");
        return 0;
    }
    for(i=0;i<n-1;i++)
    {
        scanf("%d",&tmp2);
        a[i]=tmp2-tmp;
        tmp=tmp2;
    }
    scanf("%d",&tmp);
    for(i=0;i<m-1;i++)
    {
        scanf("%d",&tmp2);
        s[i]=tmp2-tmp;
        tmp=tmp2;
    }
    //1
    KMP(1);
    //2
    for(i=0;i<alen/2;i++)
      swap(a[i],a[alen-1-i]);
    KMP(1);
    //3
    memset(a,0,sizeof(a));memset(f,0,sizeof(f));
    KMP(2);
    printf("%d",ans);
    return 0;
}