原题链接:
http://acm.zzuli.edu.cn/problem.php?id=2264
题意:
有一个长为n的数组a和一个长为m的数组b,询问在A中有多少段连续的长为m的子序列满足A[k],[Ak+1],…,A[k+m-1]使得对于任意1<=i, j<=m满足A[k+i-1]-B[i]=A[k+j-1]-B[j]
思路:
变形一下就是对于a的[l,r]中任意的i,j:
a[l+i-1]-a[l+j-1]=b[i]-b[j]
那么实际上就是求b的差分数组在a的差分数组中出现了几次
求b串在a串中出现了几次,用kmp
模板:
https://demooo.top/algorithm/%E5%AD%97%E7%AC%A6%E4%B8%B2/kmp.html
#include<bits/stdc++.h>
using namespace std;
#define int long long
#include<vector>
const int N=1e6+10;
int a[N];
int b[N];
int s2[N],s1[N];
int p[N];
int n,m;
void pre(){
int j=0;
p[1]=0;
for(int i=1;i<m;i++){
while(j&&s2[i+1]!=s2[j+1]) j=p[j];
if(s2[i+1]==s2[j+1])j++;
p[i+1]=j;
}
}
signed main(){
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=m;i++)cin>>b[i];
for(int i=1;i<n;i++) s1[i]=a[i+1]-a[i];
for(int i=1;i<m;i++) s2[i]=b[i+1]-b[i];
n--;
m--;
// for(int i=0;i<n;i++)cout<<s1[i]<<" ";
// cout<<endl;
// for(int i=0;i<m;i++) cout<<s2[i]<<" ";
// cout<<endl;
pre();
int j=0;
int con=0;
for(int i=0;i<n;i++){
while(j&&s1[i+1]!=s2[j+1])j=p[j];
if(s1[i+1]==s2[j+1])j++;
if(j==m){
con++;
j=p[j];//继续寻找匹配(可重叠,不可重叠的话j=0)
}
}
cout<<con;
return 0;
}