HDOJ 1711 (用KMP解决)

hdoj 1711 Number Sequence:  http://acm.hdu.edu.cn/showproblem.php?pid=1711

此题利用KMP算法即可解决。

状态转移图的构造是KMP算法的关键，算法的思想是“用自己匹配自己”。根据f[0], f[1],f[2],f[3],...f[i-1] 递推f[i] ，

如下面的代码getFail 函数。

kmp具体分析参见：http://www.cppblog.com/oosky/archive/2006/07/06/9486.html。
注意审题，刚开始数组开小了，一直报错 wrong answer..

code：

#include <stdio.h>
#include <string.h>

#define M 10005
#define N 1000005

int a[N],b[M],f[M];
int na,nb;// a,b实际输入数字的个数


void getFail()// 构造转移图
{
	memset(f, 0, sizeof(f));// 递推边界初值
	int i,j;
	for(i=1; i<nb; i++)
	{
		j=f[i]; // **
		while(j && b[i]!=b[j])
			j=f[j];
		f[i+1]= b[i]==b[j]?j+1:0;
	}
}
int find()
{
	getFail();
	int i,j=0;//  j 当前节点编号
	for(i=0; i < na; i++)
	{
		while(j && a[i]!=b[j])// 顺着失配边走，直到可以匹配
			j=f[j];
		if(a[i]==b[j])
			j++;
		if(j==nb)// find it
			return i-nb+2;
	}
	return -1;
}

int main()
{
	//freopen("in.txt", "r", stdin);
	int i,t;// 
	scanf("%d", &t);
	while(t--)
	{
		scanf("%d%d", &na, &nb);
		for(i=0; i<na; i++)
			scanf("%d", &a[i]);
		for(i=0; i<nb; i++)
			scanf("%d", &b[i]);
		printf("%d\n", find());
	}
	return 0;
}