【codevs3945】完美拓印 KMP

题目描述 Description

小Q获得了一个神奇的印章，这个印章宽n个单位长度，印章的其中三个棱都是直的，而另外一个方向上，对于每个单位宽度的部分，是一样直的，并且与反方向的棱平行，如下图所示。

小Q的印章上有一个不关于中心对称的图形（不一定是上图的Qrz），他现在要在一张地图上拓上印，地图上有一段个m单位长度、近似水平的边界线，但是放大到单位长度时还是有一定的高低差异，但对于单位宽度的部分，是一样直的，与水平轴线垂直，如下图所示。

小Q希望自己的印章一边的边缘能恰好地与边界线重合（不能部分重合、不能越过边界线），他现在只可以将印章旋转180度或者不旋转（这样印章可能存在有两边可以与边界线重合的情况），然后平移到适当的位置，问小Q有多少种可行的方案，两种方案不同被定义为两种方案用印章印出的图案互不重合。

输入描述 Input Description

第一行两个正整数n和m，表示印章的宽度和截取边界线的长度。

第二行n个正整数，表示印章从左到右每个单位宽度对应的两条平行线之间的距离。

第三行m个整数，表示所截取边界线从左到右每个单位宽度对应的竖直方向上的坐标。

输出描述 Output Description

一个整数，表示可行方案的种类数。

样例输入 Sample Input

5 12

3 4 4 3 2

2 4 5 5 4 3 2 2 3 3 2 1

样例输出 Sample Output

3

数据范围及提示 Data Size & Hint

对于30%的数据，有n*m≤2*107。

对于60%的数据，有n，m≤105。

对于100%的数据，有n,m≤106，所有数字的绝对值不超过109。

差分后kmp。

不知道为什么，我的j在匹配的时候会超出长度n…因为第n位是0是有意义的……

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int SZ = 1000010;
const int INF = 1000000010;

int n,m;

int nxt[SZ];

void getnxt(int s[])
{
    nxt[1] = nxt[0] = 0;
    for(int i = 1;i < n;i ++)
    {
        int j = nxt[i];
        while(j && s[i] != s[j]) j = nxt[j];
        nxt[i + 1] = s[i] == s[j] ? j + 1 : 0;
    }
}

int kmp(int s1[],int s2[])
{
    s1[n] = -INF; s2[m] = -INF;
    getnxt(s1);
    int ans = 0;
    for(int i = 0,j = 0;i < m;i ++)
    {
        while(j && s1[j] != s2[i]) j = nxt[j];
        if(s1[j] == s2[i]) j ++;
        if(j == n) ans ++;
    }
    return ans;
}



int s1[SZ],s2[SZ];
int num1[SZ],num2[SZ];
int tmp[SZ];
int main()
{
    scanf("%d%d",&n,&m);
    if(n == 1) { printf("%d\n",m * 4); return 0; }
    for(int i = 1;i <= n;i ++)
        scanf("%d",&num1[i]);
    for(int i = 1;i <= m;i ++)
        scanf("%d",&num2[i]);   

    n --; m --;
    for(int i = 0;i < m;i ++)
        s2[i] = num2[i + 2] - num2[i + 1];

    int ans = 0;

    for(int i = 0;i < n;i ++)
        s1[i] = num1[i + 2] - num1[i + 1];
    ans += kmp(s1,s2);

    for(int i = 0;i < n;i ++)
        tmp[n - i - 1] = s1[i]; 
    ans += kmp(tmp,s2);

    memset(s1,0,sizeof(s1));
    ans += kmp(s1,s2) * 2;

    printf("%d",ans);
    return 0;
}