HDU 1711 Number Sequence(KMP:找模板第一次出现的位置)

HDU 1711 Number Sequence(KMP:找模板第一次出现的位置)

http://acm.hdu.edu.cn/showproblem.php?pid=1711

题意:

        给你T串和模板P串,输出P第一次出现的位置.

分析:

        简单KMP,直接用模板,详见刘汝佳:训练指南.

AC代码:

<span style="font-size:18px;">#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int MAXN=1000000+100;
const int MAXM=10000+100;
int T[MAXN],P[MAXM],f[MAXM],f2[MAXM];
int n,m;
void getFail(int *P,int *f)//必须用f和f2两个数组,要不然不好计算优化指针,比如串aaaab的优化指针是000030
{
    f[0]=f[1]=0;//保存优化之后的失配指针
    f2[0]=f2[1]=0;//f2保存未优化的适配指针
    for(int i=1;i<m;i++)
    {
        int j=f2[i];
        while(j && P[i]!=P[j]) j=f2[j];
        f2[i+1] = f[i+1] = (P[i]==P[j])?j+1:0;
        if(f[i+1]==j+1 && P[i+1]==P[j+1]) f[i+1]=f[j+1];
    }
}
int find(int *T,int *P,int *f)
{
    getFail(P,f);
    int j=0;
    for(int i=0;i<n;i++)
    {
        while(j && T[i]!=P[j]) j=f[j];
        if(T[i]==P[j]) j++;
        if(j==m) return i-m+1+1;//这里多加了个1,因为数组下标从0开始,而题目中下标从1开始
    }
    return -1;
}

int main()
{
    int k;
    scanf("%d",&k);
    while(k--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            scanf("%d",&T[i]);
        T[n]=1e8;
        for(int i=0;i<m;i++)
            scanf("%d",&P[i]);
        P[m]=1e8;
        int ans = find(T,P,f);
        printf("%d\n",ans);
    }
    return 0;
}
</span>