hpuoj1697 CZY找句子【KMP】

1697: CZY找句子

时间限制: 1 Sec   内存限制: 128 MB
提交: 99   解决: 45
[ 提交][ 状态][ 讨论版]

题目描述

YY喜欢读书，而且总是喜欢从书上摘取好的句子或词语，但是抄完之后她变忘了在哪，于是她就请CZY帮她，CZY感觉这我多我可咋找啊！
于是又来麻烦聪明的你，为了方便CZY早已用高端译码机器把句子变成的数字，他想让你告诉他句子在文章中出现的位置。（位置从1开始）

输入

输入包括多组测试数据，每组测试数据包含三行，第一行包含两个整数n（1<=n<=1000000）,m(1<=m<=1000),分别代表文章的长度，和所摘取的句子的长度，
第二行包含n个整数表示文章，第三行包括m各整数表示句子

输出

输出所摘取的句子在文章中的位置

样例输入

13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3

样例输出

6

#include<stdio.h>
#include<string.h>
const int maxn = 1010;
int p[maxn];
int b[maxn],a[maxn*1000];
int n,m,ans;
void getp()
{
	int i=0,j=-1;
	p[i]=j;
	while(i<m)
	{
		if(j==-1||b[i]==b[j])
		{
			i++,j++;
			p[i]=j;
		}
		else j=p[j];
	}
}
void kmp()
{
	getp();
	int i=0,j=0;
	while(i<n)
	{
		if(j==-1||a[i]==b[j])
		{
			i++;
			j++;
		}
		else j=p[j];
		if(j==m)
		   ans=i+1-m;
	}
}
int main()
{
	int i,j;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
	    for(i=0;i<n;i++)
	       scanf("%d",&a[i]);
	    for(j=0;j<m;j++)
	       scanf("%d",&b[j]);
	    kmp();
	    printf("%d\n",ans);
	}
}