HDU 1711 Number Sequence （KMP算法）

#include <iostream>
using namespace std ;

int a[1000000+100];
int b[10000+100];
int nextval[10000+100];


void get_nextval(int b[],int nextval[],int m)
{
int i=1,j=0;
nextval[1]=0;
while(i<=m)
{
if(j==0 || b[i] == b[j])
{
i++;
j++;
// nextval[i]=j;
if(b[i]!=b[j]) nextval[i]=j;//防止出现模式串为"aaaab" next数组为{0,1,2,3,4}的情况出现
else nextval[i]=nextval[j];// 现在的nextval数组为{0,0,0,0,4}， 优化时间 
}
else
{
j=nextval[j];
}
}
}
int KPM(int a[],int b[],int n,int m)
{
int i=1,j=1;
while(i<=n && j<=m)
{
if(j==0 || a[i]==b[j])
{
i++;
j++;
}
else 
{
j=nextval[j];
}
}
if(j>m) return i-m;
else return -1;
}
int main()
{
int T;
cin >> T;


while(T--)
{
int n , m ;

cin >> n >> m ;


int i; 
for(i=1;i<=n;i++)
{
cin >> a[i];
}
for(i=1;i<=m;i++)
{
cin >> b[i];
}


get_nextval(b,nextval,m);
int k=KPM(a,b,n,m);


cout << k << endl ;


}
}

/*
3
10 5
2 2 3 4 5 6 1 2 3 4
2 3 4 5 6
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 2 1

*/