【kmp算法】KMP找子串第一次出现的位置

hdu1711

题意：kmp水题

思路：只需要第一次匹配成功时标记一下当前位置，输出开始匹配成功的开始匹配的位置即可。

【套路】

#include<cstdio>
using namespace std;

const int Max=1000001,Lar=10001;
int tem[Max];
int mar[Lar];
int next[Lar];

void getnext(int n)
{
	int i=0,j=-1;
	next[i]=-1;
	while(i<n) {
		if(j==-1||mar[i]==mar[j]) {
			next[++i]=++j;
		}
		else j=next[j];
	}
}

int kmp(int pos,int m,int n)
{
	int i=pos,j=0,ans,flag=0;
	while(i<n) {
		if(mar[j]==tem[i]||j==-1) {
			++i,++j;
		}
		else j=next[j];
		if(j==m) {
			ans=i;
			flag=1;
			break;
		}
	}
	if(flag) return ans-m+1;
	else return -1;
}

int main()
{
	int T,n,m;
	scanf("%d",&T);
	while(T--) {
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++) scanf("%d",&tem[i]);
		for(int i=0;i<m;i++) scanf("%d",&mar[i]);
		getnext(m);
		printf("%d\n",kmp(0,m,n));
	}
	return 0;
}