HDU 1711 Number Sequence（KMP模板题）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1711

题意

        给你两个序列，让你求B序列在A序列第一次出现（完全相同）的下标，若A中不存在B一样的序列则输出-1。

思路

        很明显如果采用暴力比较的话会TLE，所以采用KMP算法匹配字符串，先求出next数组，然后进行匹配，时间复杂度为$O(m+n)$。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;

int a[maxn],b[maxn];
int Next[maxn];
int n,m;

void getNext()
{
	Next[0]=-1;
	int i=0,j=-1;
	while(i<m){
		if(j==-1||b[i]==b[j]){
			i++;
			j++;
			Next[i]=j;
		}
		else{
			j=Next[j];
		}
	}
	return;
}

int kmp()
{
	int cnt=0;
	int i=0,j=0;
	while(i<n){
		if(j==-1||a[i]==b[j]){
			i++;
			j++;
		}
		else{
			j=Next[j];
		}
		if(j==m){
			return i-m+1;
		}
	}
	return -1;
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--){
		scanf("%d %d",&n,&m);
		for(int i=0;i<n;i++) scanf("%d",&a[i]);
		for(int i=0;i<m;i++) scanf("%d",&b[i]);
		getNext();
		printf("%d\n",kmp());
	}
	return 0;
}