hdu1711

模板題:

#include <iostream>
#include <stdio.h>
using namespace std;
int NEXT[10010];
int s[1000010],p[10010];
int n,m;
void GetNEXTval()
{
    int pLen = m;
    NEXT[0] = -1;
    int k = -1;
    int j = 0;
    while (j < pLen - 1)
    {
        //p[k]表示前綴,p[j]表示後綴
        if (k == -1 || p[j] == p[k])
        {
            ++j;
            ++k;
            //較之前NEXT數組求法,改動在下面4行
            if (p[j] != p[k])
                NEXT[j] = k;   //之前只有這一行
            else
                //因爲不能出現p[j] = p[ NEXT[j ]],所以當出現時需要繼續遞歸,k = NEXT[k] = NEXT[NEXT[k]]
                NEXT[j] = NEXT[k];
        }
        else
        {
            k = NEXT[k];
        }
    }
}
int KmpSearch()
{
    int i = 0;
    int j = 0;
    int sLen = n;
    int pLen = m;
    while (i < sLen && j < pLen)
    {
        //①如果j = -1,或者當前字符匹配成功(即S[i] == P[j]),都令i++,j++
        if (j == -1 || s[i] == p[j])
        {
            i++;
            j++;
        }
        else
        {
            //②如果j != -1,且當前字符匹配失敗(即S[i] != P[j]),則令 i 不變,j = NEXT[j]
            //NEXT[j]即爲j所對應的NEXT值
            j = NEXT[j];
        }
    }
    if (j == pLen)
        return i - j;
    else
        return -1;
}
int main()
{
    int t;cin>>t;
    while(t--)
    {
        cin>>n>>m;
        for(int i=0;i<n;i++)
            scanf("%d",&s[i]);
        for(int i=0;i<m;i++)
            scanf("%d",&p[i]);
         GetNEXTval();
    int ans=KmpSearch();
    if(ans==-1) cout<<-1<<endl;
    else cout<<ans+1<<endl;
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章