b站王道考研-数据结构4.2_4KMP优化算法实现-代码

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std; 

//字符串存储结构，序号从1开始
#define MaxLen 255
typedef struct{
	char ch[MaxLen];
	int length;
}SString;

//nextval数组求解
void get_nextval(SString T,int *nextval){
	int i=1,j=0;
	nextval[1]=0;
	while(i<T.length){
		if(j==0||T.ch[i]==T.ch[j]){
			++i;++j;
			if(T.ch[i]!=T.ch[j])nextval[i]=j;//相对于next数组的优化。
			else nextval[i]=nextval[j]; 
		}
		else
			j=nextval[j];
	} 
} 

//KMP算法
int Index_KMP(SString S,SString T,int nextval[]){
	int i=1,j=1;
	while(i<=S.length&&j<=T.length){
		if(j==0||S.ch[i]==T.ch[j]){
			++i;++j;//匹配成功，以及j=0时后移两个指针 
		}
		else
			j=nextval[j]; //字符不匹配，i不动，模式串后移。 
	}
	if(j>T.length)
		return i-T.length;//匹配成功
	else
		return 0; 
} 

int main(void){
	SString S;
	SString T;
	scanf("%s",S.ch+1);//使得字符串从下标1开始输入 
	scanf("%s",T.ch+1);
	S.length = strlen(S.ch)-1;
	T.length = strlen(T.ch)-1;
	int nextval[T.length];//获取nextval数组 
	get_nextval(T,nextval);
	printf("匹配位置：%d",Index_KMP(S,T,nextval));
}