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;
}