kmp算法代码

# include <iostream>
# include <string>
using namespace std;
void get_next(string a,int *next)
{
    int i=-1,j=0;
    next[0]=-1;//方便起见将next[0]赋值为-1；
    while(j<a.length()-1){
        if(-1==i||a[i]==a[j]){
            ++i;
            ++j;
            if(a[i]!=a[j])
               next[j]=i;
            //当前i即代表j前的字串的重叠个数，
            else
                next[j]=next[i];
        }
        else{
            i=next[i];
            //当前字符不相等，i向前回溯，寻找是否有更短的重复子串
        }
    }
}
int KMP(string a,string b)
{
    int i=0,j=0;
    int next[1000];
    get_next(b,next);
    while(i+1<a.length()+1&&j+1<b.length()+1)
    {
        if(a[i]==b[j]||j==-1){
            ++i;
            ++j;
        }//相等或者j回溯到首个字符向下+1
        else{
            j=next[j];//寻找重叠字串位置继续向下寻找
        }
        
    }
    if(j>=b.length())
        return i-b.length();//返回字符串重叠的起始位置
    else
        return -1;
}
int main(){
    string a,b;
    getline(cin,a);
    getline(cin,b);
    cout<<KMP(a,b)<<endl;
    return 0;
}