HDU 1711 Number Sequence KMP模板题

题目：

传送门

思路：

KMP模板题目

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=1e6+5;
const int maxm=1e4+5;
int n,m,t;
int a[maxn],b[maxm];
int p[maxm];
void prefix_table()
{
    p[0]=0;
    int len=0;
    int i=1;
    while (i<m)
    {
        if(b[len]==b[i])
        {
            len++;
            p[i]=len;
            i++;
        }
        else
        {
            if(len>0)
            {
                len=p[len-1];
            }
            else
            {
                p[i]=0;
                i++;
            }
        }
    }
    for (int j=m-1;j>0;j--)
        p[j]=p[j-1];
    p[0]=-1;
}
int Judge ()
{
    int i=0,j=0;
    while (i<n)
    {
        if(j==m-1&&b[j]==a[i])
        {
            return i-j+1;
        }
        if(a[i]==b[j])
        {
            i++;j++;
        }
        else
        {
            j=p[j];
            if(j==-1)
            {
                i++; j++;
            }
        }
    }
    return -1;
}
int main()
{
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d%d",&n,&m);
        for (int i=0;i<n;i++)
            scanf("%d",&a[i]);
        for (int j=0;j<m;j++)
            scanf("%d",&b[j]);
        prefix_table();
        printf("%d\n",Judge());
    }
    return 0;
}