hdu1711Number Sequence kmp

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
int n,m;
int st[1100000],st2[1100000],next[1100000];
void getnext()
{
    next[0]=0;
    for(int i=1;i<n;i++)
    {
        int j=next[i];
        while(j&&st[i]!=st[j])
        {
            j=next[j];
        }
        if(st[i]==st[j])
        {
            next[i+1]=j+1;
        }
        else
        {
            next[i+1]=0;
        }
    }
}
int cal()
{
    int j=0;
    for(int i=0;i<m;i++)
    {
        while(j>0&&st2[i]!=st[j])
        {
            j=next[j];
        }
        if(st2[i]==st[j])
        {
            j++;
        }
        if(j==n)
        {
            return i-n+2;//输入从1开始
        }
    }
    return -1;
}
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d%d",&m,&n);
        for(int j=0;j<m;j++)
        {
            scanf("%d",&st2[j]);
        }
        for(int i=0;i<n;i++)
        {
            scanf("%d",&st[i]);
        }
        getnext();
        printf("%d\n",cal());
    }
    return 0;
}