codevs 3945 完美拓印 (KMP)

题目大意：给你一个神奇的印章，他左右下三个面都是直的，上面是凸凹不平的面(凸凹都平行于别的面)。然后给你一个轮廓线，如果一个面能与轮廓线完全重合，可以把印章的这个沿着轮廓线拓印，求所有的拓印方案。

把轮廓线和印章相邻两个高度打个查分，然后KMP匹配一下就行了。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#define N 10010
#define mod 1000000007
#define ui unsigned int
#define ll long long
#define dd double
#define idx(x) x-'a'+1
using namespace std;
//re
int T,n,m,cnt;
int nxt[3][N];
ui f[N];
int a[N],b[N],s[N],t[3][N];
void get_kmp(int s1[],int p)
{
    int i=1,j=0;
    nxt[p][1]=0;
    while(i<m)
    {
        if(j==0||s1[i]==s1[j])
        {
            i++;
            j++;
            nxt[p][i]=j;
        }else{
            j=nxt[p][j];
        }
    }
}
int KMP(int s1[],int s2[],int p)
{
    int i=1,j=1,cnt=0;
    while(i<n)
    {
        if(j==0||s1[i]==s2[j])
        {
            i++;
            j++;
        }else{
            j=nxt[p][j];
        }
        if(j==m)
        {
            cnt++;
            j=nxt[p][j];
        }
    }
    return cnt;
}


int main()
{
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++) scanf("%d",&b[i]);
    for(int j=1;j<=n;j++) scanf("%d",&a[j]);
    if(m==1) {printf("%d\n",n*4);return 0;}
    for(int i=1;i<m;i++) t[0][i]=b[i+1]-b[i],t[2][i]=0;
    for(int i=1;i<m;i++) t[1][i]=t[0][m-i];
    for(int i=1;i<n;i++) s[i]=a[i+1]-a[i];
    get_kmp(t[0],0),get_kmp(t[1],1),get_kmp(t[2],2);
    int ans=0;
    ans+=KMP(s,t[0],0);
    ans+=KMP(s,t[1],1);
    ans+=2*KMP(s,t[2],2);
    printf("%d\n",ans);
    return 0;
}