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;
}
}