hdu 1711 Number Sequence（KMP模板题）

我的第一道KMP。

把两个数列分别当成KMP算法中的模式串和目标串，这道题就变成了一个KMP算法模板题。

 

#include<stdio.h>
#include<string.h>
#define N 1000005
#define M 10005
int a[N],b[M];
int next[M];
int n,m;
void setNext()
{
	int i,j;
	i=0;
	j=-1;
	next[i]=j;
	while(i<m)
	{
		if(j==-1||b[i]==b[j])
		{
			i++;
			j++;
			next[i]=j;
		}
		else
			j=next[j];
	}
	return ;
}
int KMP()
{
	int i,j;
	i=j=0;
	setNext();
	while(i<n)
	{
		if(j==-1||a[i]==b[j])
		{
			i++;
			j++;
			if(j==m)
				return i-m+1;
		}
		else
			j=next[j];
	}
	return -1;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&m);
		int i;
		for(i=0;i<n;i++)
			scanf("%d",&a[i]);
		for(i=0;i<m;i++)
			scanf("%d",&b[i]);
		int temp;
		temp=KMP();
		printf("%d\n",temp);
	}
	return 0;
}