HDU 1711

字符串匹配KMP算法

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

这题。。。。没有什么说的，KMP基础题，匹配成功后退出并返回 i-j+1的值。

#include"iostream"
#include"cstdio"
#include"cstring"
using namespace std;
bool flag;
int n,m,nexts[10005],num[1000005],num2[100005];
void get_nexts(int s[],int nexts[])
{
    int i=0,k=-1;
    nexts[0]=-1;
    while(i<m)
    {
        if(k==-1||s[i]==s[k])
        {
            i++;
            k++;
            if(s[i]==s[k])
                nexts[i]=nexts[k];
            else
                nexts[i]=k;     //如果k==-1且 s[i] != s[k] 则该点nexts==-1;
        }
        else
            k=nexts[k];
    }
}
int kmp(int s1[],int s2[])
{
    int i=0,j=0,id=-1;
    while(i<n&&j<m)    //这里表明 匹配串匹配完成后就会退出了。
    {
        if(j==-1||s1[i]==s2[j])
        {
            i++;
            j++;
        }
        else
            j=nexts[j];
    }
    if(j>=m)
        id=i-j+1;
    return id;
}
int main()
{
    int t;

    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&m);

        for(int i=0;i<n;i++)
            scanf("%d",&num[i]);
        for(int j=0;j<m;j++)
            scanf("%d",&num2[j]);
        num[n]=10000000;
        num2[m]=1000000;

        get_nexts(num2,nexts);
        int ans=kmp(num,num2);
        printf("%d\n",ans);
    }
    return 0;
}