【HDU1711】Number Sequence

题面

大致题意：

给定两个数列A，B，长度分别为N和M
求出 满足 Ak=B1 ,Ak+1=B2……Ak+M-1=Bm
的最小k值
如果有多个k值输出最小的一个

题解

KMP裸题
直接计算B数列的next值KMP匹配即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAX 1000100
int nt[MAX],A[MAX],B[MAX],N,M;
inline int read()
{
      register int x=0,t=1;
      register char ch=getchar();
      while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
      if(ch=='-'){t=-1;ch=getchar();}
      while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
      return x*t;
}
void GetNext()
{
      nt[1]=0;
      for(int i=2;i<=M;++i)
      {
              int t=nt[i-1];
              while(B[t+1]!=B[i]&&t)
                    t=nt[t-1];
              if(t==0)
              {
                    if(B[1]==B[i]) nt[i]=1;
                    else nt[i]=0;
              }
              else
                    nt[i]=t+1;
      }
}
int KMP()
{
      int i=1,j=1;
      while(j<=N)
      {
             if(B[i]==A[j])
             {
                   ++i;++j;
             }
             else
             {
                   if(i==1)j++;
                   else
                       i=nt[i-1]+1;
             }
             if(i>M)
                   return j-M;
      }
      return -1;
}
int main()
{
      int T=read();
      while(T--)
      {
             N=read();M=read();
             for(int i=1;i<=N;++i)
                A[i]=read();
             for(int j=1;j<=M;++j)
                B[j]=read();
             GetNext();
             printf("%d\n",KMP());
      }
      return 0;
}