首页 > 代码库 > 字符串匹配 - KMP算法

字符串匹配 - KMP算法

首先大致的学习一下有限自动机字符匹配算法,然后在讨论KMP算法。

有限自动机

一个有限自动机M是一个五元组(Q,q0,A,Σ,δ),其中:

  • Q是状态的集合,
  • q0∈Q是初始状态,
  • A是Q的字集,是一个接受状态集合,
  • Σ是一个有限的输入字母表,
  • δ是一个从Q×Σ到Q的函数,叫做转移函数。

下面定义几个相关函数:

  • φ(w)是M在扫描字符串w后终止时的状态。函数φ有下列递归关系定义:φ(ε) = q0,φ(wa) = δ(φ(w),a),
  • σ(x)是x的后缀中,关于P的最长前缀的长度。

字符串匹配自动机

来回顾一下朴素算法。给定下面两个字符串,模式串P,和匹配串T。

i012345678910
Pababaca    
Tabababacaba

当第一次匹配时,i=0,但是扫描到i=5的时候,字符串不在匹配。此时另i=1,重新匹配。这就是朴素算法需要改进的地方。当i=5的时候,观察表格发现P[0...3]=T[2...5],此时如果能够匹配T[5+1]和P[3+1]就不需要从i=2开始扫描了,效率就大大的提升了,这样匹配的时间复杂度就只有O(n)了。这里P[0...3]叫做P的前缀,T[2...5]叫做T5的后缀。此时σ(T5) = 3。这样在自动机的操作中,如果每次状态转移都能够保证:

        φ(Ti)=σ(Ti)

那么就可以保证最终的正确匹配。下面来做简单的推理:

根据φ(x)的定义,有φ(Tia) = δ(φ(Ti),a),其中a为任意字母;

由φ(Ti)=σ(Ti),可以得到φ(Tia)=σ(Tia) = q,即φ(Tia)=σ(Pqa);

综上,δ(φ(Ti),a)=σ(Pqa),,可以得到一个状态转移函数δ(q,a)=σ(Pqa)。这样就可以做出一个正确的状态转移图,然后就可以匹配字符串了。

用文字来描述一下:在自动机中,状态q就是Ti的后缀在P的最长前缀的长度。这样每次能够满足这个条件,就能够保证算法的正确进行。这里,在《算法导论》中有详细的数学证明。

KMP算法

KMP算法不建立一个有限自动机,但是必须要构建一个前缀函数,这里就叫做前缀数组吧。模式P和自己先匹配,得到前缀数组。前缀数组其实保存的就是自动机中的σ(x)的值。这样预处理的时间复杂度和自动机比就减少了很多。

预处理

给定模式P:

i0123456789
Pababababca
next0012345601

这里Pi[next[i]]表示的是Pi的关于P的最长后缀,P[i]表示P关于Pi的前缀。

当i=0时:

P0和P比较,P0[0] != P[0],所以next[0]=0;

当i=1时:

P1和P比较,P1[0] != P[1],所以next[1]=0;

当i=2时:

P2和P比较,P2[0] = P[2],所以next[2]=1;

当i=3时:

P3和P比较,P3[1] = P[3],所以next[3] = 2;

如此这般,就可以求得next数组了。一般算法描述数组都是从1开始,但是写代码的时候,数组是从下标0开始的,所以上面的next数组的每一个值都应该减一。next[i]=-1表示没有前缀匹配。这样在写代码的时候,应该是这样的:

i0123456789
Pababababca
next-1-1012345-10

当i=0时,初始化next[0] = -1;

当i=1时,(P1[next[0]+1] = a) != (P[1] = b),next[1] = -1;

当i=2时,(P2[next[1]+1] = a) != (P[2] = a),next[2] = 0;

当i=3时,(P3[next[2]+1] = b) !=(P[3] = b),next[3] = 1;

...

这样就不难发现next数组的作用了,记录了当前的σ(Pi)。Pi[next[i]+1] = P[i],就表示Pi最长前缀加一个字母和P的后缀加一个字母是否匹配。此时有两种情况:

  • Pi[next[i]+1] = P[i],这个时候σ(Pi+1) = σ(Pi) + 1,继续
  • Pi[next[i]+1] != P[i],这个时候没有直接从Pi[0]是不是等于P[i]扫描,而是从Pi[next[next[i]]+1]开始扫描。因为目前一定可以保证Pi[next[next[i]]是P的一个前缀。

下面是C代码的实现的求next数组:

void get_next(char *P, int next[],int len){	printf("len=%d\n",len);	next[0] = -1;	int q = -1;	int i;	for(i = 1; i < len; i++) {		while(q > 0 && P[q+1] != P[i]) { /* 判断P[q+1]适合等于P[i] */			q = next[q]; /* 如果不相等, 一直找到满足条件的最长后缀 */		}		if(P[q+1] == P[i]) q++; /* 如果相等,那么很好,继续... */		next[i] = q;			}}

 匹配

当求出next数组后,就可以进行字符串匹配了。匹配的方法和求next的方法相识。下面是完整的代码:

/*************************************************************************    > File Name: KMP.c    > Author: mr_zys    > Mail: 247629929@163.com     > Created Time: 2014年10月09日 星期四 14时48分30秒 ************************************************************************/#include<stdio.h>#include<string.h>#define maxn 100int next[maxn];char P[maxn],T[maxn];void get_next(char *P, int next[],int len){	printf("len=%d\n",len);	next[0] = -1;	int q = -1;	int i;	for(i = 1; i < len; i++) {		while(q > 0 && P[q+1] != P[i]) { /* 判断P[q+1]适合等于P[i] */			q = next[q]; /* 如果不相等, 一直找到满足条件的最长后缀 */		}		if(P[q+1] == P[i]) q++; /* 如果相等,那么很好,继续... */		next[i] = q;			}}void KMP(char *P, char *T){	int len_P = strlen(P);	int len_T = strlen(T);	int j = -1;	int i;	for(i = 0; i < len_T; i++) {		while(j > -1 && T[i] != P[j+1]) {			j = next[j];		}		if(P[j+1] == T[i]) {			j++;			//printf("%d %d\n",j,i);		}		if(j == len_P-1){			printf("在%d处开始匹配\n",i-len_P+1);			j = next[j];		}	}}int main(){	printf("input the string P:\n");	scanf("%s",P);	printf("input the string T:\n");	scanf("%s",T);	printf("%s\n",P);	get_next(P,next,strlen(P));	int i;	for(i = 0; i < strlen(P); i++) {		printf("(%d)",next[i]);	}	printf("\n");	KMP(P,T);	return 0;}

 可能,中间有些表述不清,求指正哈!

-end-

 

字符串匹配 - KMP算法