KMP算法 C++版

#include <iostream>
#include <cstring>
using namespace std;

void makeNext(char* T, int next[]){
    next[0] = -1;
    next[1] = 0;
    int k = 0;//字符串最大相同前后缀长度
    int size = strlen(T);
    for(int i=2; i<size; ++i){
        while(k>0 && T[i-1]!=T[k]){
            //k=0表示字符串没有相同的前后缀，所以直接比较首尾即可，不用参与此循环
            //如果T[i-1] != T[k]，则看看k之前字符串有没有相同前后缀，即k=next[k]，
            //如果有则k跳到相同前缀后一个字符，如果没有则k跳至首部 
            k = next[k];
        }
        if(T[i-1] == T[k]){
            //如果相同则在前一个next基础上加一 
            ++k;
        }
        next[i] = k;
    }
}

void makeNextval(char* T, int next[], int nextval[]){
    nextval[0] = -1;
    int t_size = strlen(T);
    for(int i=1; i<t_size; ++i){
        if(T[i] == T[next[i]]){
            nextval[i] = nextval[next[i]];
        }
        else{
            nextval[i] = next[i];
        }
    }
}

int KMP(char* S, char* T, int next[]){
    int s_size = strlen(S);
    int t_size = strlen(T);
    int i = 0, j = 0;
    while(i<s_size && j<t_size){
        if(j==-1 || S[i] == T[j]){
            ++i;
            ++j;
        }
        else if(j==0){
            ++i;
        } 
        else{
            j = next[j];
        }
    }
    if(j==t_size)
        return i-t_size;
    else
        return -1;
}

int main(){
    int next[10];
    int nextval[10];
    char S[] = "ababcabe";
    char T[] = "abc";
    makeNext(T, next);
    makeNextval(T, next, nextval);
    cout << KMP(S, T, next) <<endl;
    cout << KMP(S, T, nextval) <<endl;
    return 0;
}