KMP基础

kmp代码

 

pre[i]表示[0, i ]段的子串（闭区间），最长匹配前缀是[0, pre[i] ]。当指针 j 失配时，i 不变，j 移动到 pre[ j - 1 ] + 1 继续匹配。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=10000;
char A[MAXN], B[MANX];
int *pre=new int[strlen(B)];

void getPre()
{
    pre[0]=-1;
    for(int i=0,k=-1;i<strlen(B);i++)
    {
        while(k>=0&&tex[i]!=tex[k+1])k=pre[k];
        if(tex[k+1]==tex[i])k++;
        pre[i]=k;
    }        
}

int findPos()
{
    int i=0,j=0,la=strlen(A),lb=strlen(B);
    while(i<la&&j<lb)
    {
        if(A[i]==B[j])i++,j++;
        else j=pre[j-1]+1;
    }
    if(j==lb)return i-lb;
    else return -1;
}

题目：洛谷3375  ： 给出两个字符串s1和s2，其中s2为s1的子串，求出s2在s1中所有出现的位置。

#include<bits/stdc++.h>
using namespace std;

const int MAXN = 1e6 + 5;
char a[MAXN], b[MAXN];

int main()
{
	freopen("C:\\Users\\DELL\\Desktop\\testdata.in", "r", stdin);
	freopen("C:\\Users\\DELL\\Desktop\\outdata.txt", "w", stdout);

	scanf("%s", a);
	scanf("%s", b);
	int la = strlen(a), lb = strlen(b);//字符串很长时，如果后面每次都单独求，会让程序慢很
    //很多秒，没法过
	//cout << "len: " << strlen(a) << " " << strlen(b) << endl;
	int *pre = new int[lb];
	pre[0] = -1;
	for (int i = 1, k = -1; i<lb; i++)
	{
		while (k >= 0 && b[k + 1] != b[i])k = pre[k];
		if (b[k + 1] == b[i])k++;
		pre[i] = k;
	}
	
	int i = 0, j = 0;
	while (i<la)
	{
		while (i<la && j<lb)
		{
			if (a[i] == b[j])i++, j++;
			else if(j) j = pre[j - 1] + 1;
			else i++;
		}
		if (j == lb)
		{
			cout << i - lb + 1 << endl;
			//j = 0;
			j = pre[j-1]+1;//寻找下一个匹配位置，而不必前移i，重置j
		}
		else break;
		
	}

	system("pause");
	return 0;
}

next数组也可以用于求最小循环节，此处给出一道与最小循环节相关题目

#include<bits/stdc++.h>
using namespace std;

const int MAXN = 1e6 + 5;
char str[MAXN];
int snext[MAXN], len;


void getNext();
int main()
{
	while (scanf("%s", str) == 1)
	{
		if (str[0] == '.'&&str[1] == '\0')break;
		len = strlen(str);
		getNext();
        //注意以下几行的逻辑
        //当没有匹配时，next=-1，+1刚好为0，特殊值-1的处理不会冲突，不需要额外处理
        //特殊边界值的选取保持与普通纸逻辑一致，往往可以减少边界条件处理工作量
		int t = snext[len - 1] + 1;
		if (2 * t < len)cout << 1 << endl;
		else cout << len / (len - t) << endl;
	}


	system("pause");
	return 0;
}

void getNext()
{
	memset(snext, 0, sizeof(snext));
	snext[0] = -1;
	for (int i = 1, k = -1; i < len; i++)
	{
		while (k >= 0 && str[i] != str[k + 1])k = snext[k];
		if (str[i] == str[k + 1])k++;
		snext[i] = k;
	}
}