HDUNumber Sequence(KMP)

妖精的绣舞 提交于 2019-12-08 12:55:39

传送门

题目大意:b在a第一次出现的位置

题解:KMP

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000008
#define M 10009
using namespace std;

int T;

int a[N],b[M];

int nxt[M];

int lena,lenb;

void getnext()
{
    memset(nxt,0,sizeof(nxt));
    for(int i=2,j=0;i<=lenb;i++)
    {
        while(b[i]!=b[j+1]&&j) j=nxt[j];
        if(b[i]==b[j+1]) nxt[i]=++j;
    }
}

int KMP()
{
    getnext();
//    for(int i=0;i<=lenb;i++) cout<<i<<"---"<<nxt[i]<<endl;
    for(int i=1,k=0;i<=lena;i++)
    {
        for(;a[i]!=b[k+1]&&k;k=nxt[k]);
        if(a[i]==b[k+1])++k;
        if(k==lenb) return i-lenb+1;
    }
    return -1;
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&lena,&lenb);
        for(int i=1;i<=lena;i++) scanf("%d",&a[i]);
        for(int i=1;i<=lenb;i++) scanf("%d",&b[i]);
        printf("%d\n",KMP()); 
    }
    return 0;
} 

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!