题意:
给定两个序列
在a中找到一个子串 使得每个数同时加上某个整数(可以为负数)后和 b 相等
问有多少个这样的子串
思路:
我们从变量中寻找不变量,对两个序列作前缀差,这样就把变量变成了不变量,然后跑kmp模板,巧妙!
#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int Next[maxn];
void getnext(int *ptr,int n)
{
int i,k;
k=-1;
Next[0]=-1;
i=0;
while(i<n)
{
if(k==-1 || ptr[i]==ptr[k])
{
i++;
k++;
Next[i]=k;
}
else k=Next[k];
}
}
int kmp(int *a,int *b,int len1,int len2)//查找子串出现个数
{
int i=0,j=0,ans=0;
getnext(b,len2);
while(i<len1)
{
if(j==-1||a[i]==b[j])
++i,++j;
else
j=Next[j];
if(j==len2)//匹配到一个了就当没找到然后跳到next[J]
{
j=Next[j];
ans++;
}
}
return ans;
}
int s[maxn],p[maxn];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++)
{
scanf("%d",&s[i]);
if(i)s[i-1]=s[i]-s[i-1];
}
for(int i=0; i<m; i++)
{
scanf("%d",&p[i]);
if(i)p[i-1]=p[i]-p[i-1];
}
if(m==1)
{
printf("%d",n);
return 0;
}
printf("%d\n",kmp(s,p,n-1,m-1));
return 0;
}