HDU - 1711 Number Sequence（KMP）

HDU - 1711 Number Sequence

#include<cstdio>
#include<cstring>
const int N = 1000010, M = 10000;
int n,m;
int a[N],b[M],ne[M];
void get_next()
{
    for(int i=2,j=0;i<=m;i++)
	{
		while(j&&b[i]!=b[j+1]) j=ne[j];
		if(b[i]==b[j+1]) j++;
		ne[i]=j;
	}
}
int get_kmp()
{
    for(int i=1,j=0;i<=n;i++)
	{
		while(j&&a[i]!=b[j+1]) j=ne[j];
		if(a[i]==b[j+1]) j++;
		if(j==m) return i;
	}
	return -1;
}
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
    	memset(ne,0,sizeof ne); 
        scanf("%d%d",&n,&m);
    	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    	for(int i=1;i<=m;i++) scanf("%d",&b[i]);
        get_next();
    	int c=get_kmp();
    	if(c==-1) printf("-1\n");
    	else printf("%d\n",c-m+1);
    }
	return 0;
}