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;
}