zzuli 2264: sequence【kmp】

题目描述

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

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

输入

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

接下来一行n个整数，描述序列A (A_i <= 10⁹)

接下来一行m个整数，描述序列B (B_i <= 10⁹)

输出

输出一个数表示答案

样例输入

7 46 6 8 5 5 7 47 7 9 6

样例输出

2

题意：由等式变形可得：Ak+i-1 - Ak+j-1 = Bi - Bj 在处理好数组自身前后值之差后，匹配子串在主串中出现的次数。

思路：KMP模板题~~

#include<stdio.h>
#include<string.h>
const int maxn = 1000100;
int a[maxn],b[maxn],next[maxn],sum;
int n,m;

void get_next()
{
	int i,j;
	i = 0;
	j = -1;
	next[0] = -1;
	while(i < m)
	{
		while(j!=-1&&b[i]!=b[j])
			j = next[j];
		next[++i] = ++j;
	}
	return ;
}
void kmp()
{
	int i,j;
	i = j = 0;
	while(i < n)
	{
		while(j!=-1&&a[i]!=b[j])
			j = next[j];
		if(j == m-1)
		{
			sum ++;
			j = next[j];
		}
		++i;
		++j;
	}
	return;
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(next,0,sizeof(next));
		sum = 0;
		for(int i = 0; i < n; i ++)
			scanf("%d",&a[i]);
		for(int i = 0; i < m; i ++)
			scanf("%d",&b[i]);
		for(int i = 0; i < n-1; i ++)
			a[i] = a[i+1]-a[i];
		for(int i = 0; i < m-1; i ++)
			b[i] = b[i+1]-b[i];
		n -= 1;
		m -= 1;
		get_next();
		kmp();
		printf("%d\n",sum);
	}
	return 0;
}