數據結構實驗之串一：KMP簡單應用

題目描述

給定兩個字符串string1和string2，判斷string2是否爲string1的子串。

輸入

 輸入包含多組數據，每組測試數據包含兩行，第一行代表string1(長度小於1000000)，第二行代表string2（長度小於1000000），string1和string2中保證不出現空格。

輸出

 對於每組輸入數據，若string2是string1的子串，則輸出string2在string1中的位置，若不是，輸出-1。

示例輸入

abc
a
123456
45
abc
ddd

示例輸出

1
4
-1

KMP算法，時間複雜度O（n+m）,效率極高

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
char a[1001000],b[1001000];
int next[1001000];
void getnext()
{
    int i=0,j=-1;
    next[0]=-1;
    while(b[i]!='\0')
    {
        if(j==-1||b[i]==b[j])
        {
            i++;
            j++;
            next[i]=j;
        }
        else
        {
            j=next[j];
        }
    }
}

void kmp()
{
    int i=0,j=0;
    getnext();
    int k1=strlen(a);
    int k2=strlen(b);
    while(i<k1&&j<k2)
    {
        if(j==-1||a[i]==b[j])
        {
            i++;
            j++;
        }
        else
        {
            j=next[j];
        }
    }
    if(j>=k2)
    {
        printf("%d\n",i-k2+1);
    }
    else
    {
        printf("-1\n");
    }
}

int main()
{
    while(~scanf("%s%s",a,b))
    {
        kmp();
    }
    return 0;
}