题目链接:点击打开链接
题意:给你一个长度为n的序列, 让你找其中有几个长度为m的序列的形状(即高低差距)
思路:将数组预处理成相邻两数的差的形式, 然后KMP即可。 KMP的精华在于状态转移图的构造, 有点DP的思想, 也就是预处理失配之后能转移到代价尽量小的地方。
这里附上一篇KMP详解:点击打开链接
细节参见代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 200000 + 10;
int n,m,ans=0,f[maxn],T[maxn],p[maxn],a[maxn];
void getfail() {
f[0] = 0; f[1] = 0;
for(int i=1;i<m;i++) {
int j = f[i];
while(j && p[i] != p[j]) j = f[j];
f[i+1] = p[i] == p[j] ? j+1 : 0;
}
}
void _find() {
getfail();
int j = 0;
for(int i=0;i<n;i++) {
while(j && p[j] != T[i]) j = f[j];
if(p[j] == T[i]) j++;
if(j >= m) ++ans, j = f[j];
}
}
int main() {
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++) {
scanf("%d",&a[i]);
}
for(int i=0;i<n;i++) {
if(i > 0) T[i] = a[i] - a[i-1];
else T[i] = INF;
}
for(int i=0;i<m;i++) {
scanf("%d",&a[i]);
}
for(int i=0;i<m;i++) {
if(i == 0) ;
else p[i-1] = a[i] - a[i-1];
}
ans = 0;
--m;
_find();
printf("%d\n",ans);
return 0;
}