字符串查找——KMP算法

字符串查找Vol2

面试重点

2.KMP算法

设主串下标为 i，子串下标为 j，皆初始为0
代码思想：i不回退，j退到该退的位置
时间复杂度：O(n+m) //设主串长度为n，子串长度为m

//根据str计算其next数组，并返回
//这个函数是在计算-在子串中每个位置都失配的k值，并将k值存在next数组里
static int *GetNext(const char * str)
{
	int len = strlen(str);
	int *next = (int *)malloc(len*sizeof(int));
	next[0]=-1;//0下标表示是在子串初始位置，不能再退，所以置为-1
	next[1]=0;//在1下标，前面不会有相等的真子串，所以置为0

	//找到next数组的规律
	int j=1;
	int k=0;
	while(j+1<len)
	{
		if((k==-1)||str[k]==str[j])
		{
			next[++j]=++k;
			/*next[j+1]=k+1;
			j=j+1;
			k=k+1;*/
		}
		else//不相等，k回退
		{
			k=next[k];
		}
	}
	return next;
}

int KMP(const char *str,const char *sub,int pos)
{
	if(str==NULL || sub==NULL || pos<0 || pos>strlen(str))
		return -1;
	int i=pos;
	int j=0;
	int lenstr = strlen(str);
	int lensub = strlen(sub);

	int *next = GetNext(sub);
	while(i<lenstr && j<lensub)//需要比较
	{
		if((j==-1)||str[i] == sub[j])//相等继续比较
		{
			i++;
			j++;
		}
		else//不相等，i不回退，j退到该退的位置k
		{
			j=next[j];
		}
	}
	free(next);
	
	//利用子串是否遍历完成，来判断查找是否成功，注意不能利用主串判断
	if(j >= lensub)//子串遍历完成，查找成功
	{
		return i-j;
	}
	else
	{
		return -1;
	}
	return 0;
}