模式匹配-KMP演算法

/***字串匹配演算法***/
#include<cstring>
#include<iostream>
using namespace std;

#define OK 1
#define ERROR 0
#define OVERFLOW -2
typedef int Status;
#define MAXSTRLEN 255   		//使用者可在255以內定義最長串長
typedef char SString[MAXSTRLEN+1];		//0號單元存放串的長度

Status StrAssign(SString T, char *chars) { //生成一個其值等於chars的串T
	int i;
	if (strlen(chars) > MAXSTRLEN)
		return ERROR;
	else {
		T[0] = strlen(chars);
		for (i = 1; i <= T[0]; i++)
			T[i] = *(chars + i - 1);
		return OK;
	}
}
//演算法4.3　計算next函式值
void get_next(SString T, int next[])
{ //求模式串T的next函式值並存入陣列next
	int i = 1, j = 0;
	next[1] = 0;
	while (i < T[0])
		if (j == 0 || T[i] == T[j])
		{
			++i;
			++j;
			next[i] = j;
		}
		else
			j = next[j];
}//get_next

//演算法4.2　KMP演算法
int Index_KMP(SString S, SString T, int pos, int next[])
{ 	// 利用模式串T的next函式求T在主串S中第pos個字元之後的位置的KMP演算法
	//當中，T非空。1≤pos≤StrLength(S)
	int i = pos, j = 1;
	while (i <= S[0] && j <= T[0])
		if (j == 0 || S[i] == T[j]) // 繼續比較後繼字
		{
			++i;
			++j;
		}
		else
			j = next[j]; // 模式串向右移動
	if (j > T[0]) // 匹配成功
		return i - T[0];
	else
		return 0;
}//Index_KMP

int main()
{
	SString S;
	StrAssign(S,"aaabbaba") ;
	SString T;
	StrAssign(T,"abb") ;
	int *p = new int[T[0]+1]; // 生成T的next陣列
	get_next(T,p);
	cout<<"主串和子串在第"<<Index_KMP(S,T,1,p)<<"個字元處首次匹配\n";
	return 0;
}