HDU 1771 Number Sequence

Posted 2020-09-07 IKnowYou

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=1711

解题思路：简单的KMP算法实现。

实现代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAXN=1000010;
const int MAXM=10000;
void makeNext(const int P[],int next[],int m){
    int q=0;
    memset(next,0,sizeof(next));
    for(int i=1;i<m;i++){
        if(q>0&&P[q]!=P[i])
            q=next[q-1];
        if(P[q]==P[i])
            q++;
        next[i]=q;
    }
}

int kmp(const int a[],int n,const int P[],int m,int next[]){
    makeNext(P,next,m);
    for(int i=0,q=0;i<n;i++){
        while(q>0&&P[q]!=a[i])
            q=next[q-1];

        if(P[q]==a[i])
            q++;

        if(q==m)
            return i-q+1+1;
    }
    return -1;
}

int main(){
    int T;
    int a[MAXN],P[MAXM];
    int next[MAXM];
    scanf("%d",&T);
    while(T--){
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        for(int i=0;i<m;i++)
            scanf("%d",&P[i]);
        int ans=kmp(a,n,P,m,next);
        cout<<ans<<endl;
    }
}