HDU 1711 Number Sequence【KMP】【模板题】

 HDU 1711 Number Sequence

 

求母串第几位开始能和子串匹配，不能则输出-1 

#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
const int maxn = 1e6+20;
int nxt[maxn];
int m;//母 
int n;//子 
int s[maxn];//母 
int t[maxn];//子 
void getnxt()
{
    int i=1;
    int j=0;
    nxt[0]=0;
    while(i<n)
    {
        if(t[j]==t[i])
        {
            nxt[i++]=++j;
        }
        else if(j==0)
        {
            i++;
        }
        else
        {
            j=nxt[j-1];
        }
        
    }
} 
int kmp()
{
    int i=0;//母 
    int j=0;//子 
    while(i<m&&j<n)
    {
        if(s[i]==t[j])
        {
            i++;
            j++;
        }
        else if(j==0)
        {
            i++;
        }
        else
        {
            j=nxt[j-1];
        }
        
        
    }
    if(j==n) return i-n+1;
    else return -1;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&m,&n);
        for(int i=0;i<m;i++) scanf("%d",&s[i]);
        for(int i=0;i<n;i++) scanf("%d",&t[i]);
        getnxt();
        printf("%d\n",kmp());    
    }
    return 0;
}