kmp例题

题目描述

给定一个含n个数的序列A和一个含m (m<=n) 个数的序列B。

询问在A中有多少段连续的长为m的子序列Ak,Ak+1,…,Ak+m-1使得对于任意1<=i, j<=m满足Ak+i-1-Bi=Ak+j-1-Bj
输入

第一行两个整数n，m (1 <= m <= n <= 106)

接下来一行n个整数，描述序列A (Ai <= 109)

接下来一行m个整数，描述序列B (Bi <= 109)
输出

输出一个数表示答案
样例输入

7 4
6 6 8 5 5 7 4
7 7 9 6

样例输出

2

kmp算法详解
解析:直接新造两个差值数组,将两个新数组按常规kmp算即可

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define pb push_back
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rep1(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
const int N=1e6+100;
int arr1[N],app1[N];
int arr[N];
int app[N];
int nex[N];
int n,m,cnt=0;
void kmpnext()
{
    nex[0]=-1;
    int k=-1;
    for(int i=1;i<m-1;i++)
    {
        while(k>-1&&app[k+1]!=app[i])
        {
            k=nex[k];
        }
        if(app[k+1]==app[i])
        {
            k++;
        }
        nex[i]=k;
    }
}
void  kmp()
{
    kmpnext();
    int k=-1;
    for(int i=0;i<n-1;i++)
    {
        while(k>-1&&app[k+1]!=arr[i])
            k=nex[k];
        if(app[k+1]==arr[i])
            k++;
        if(k==m-2)
        {
            cnt++;
            k=nex[k];
        }
    }
}
int main()
{
    cnt=0;
    cin>>n>>m;
    rep(i,0,n)
    cin>>arr1[i];
    rep(i,0,n-1)
    arr[i]=arr1[i+1]-arr1[i];
    rep(i,0,m)
    cin>>app1[i];
    rep(i,0,m-1)
    app[i]=app1[i+1]-app1[i];
    if(m==1)
    {
        cout<<n<<endl;
        return 0;
    }
    kmp();
    cout<<cnt<<endl;
    return 0;
}