目录
题目链接:【模板】KMP - 洛谷
注意:
1、next数组是针对子串的,并未涉及母串,因此求next数组时与母串无关
2、在KMP算法中,字符串的起始下标为1,这样子方便操作,因此我们读入s母串和p子串后,要s = '0' + s; p = '0' + p
求解字符串的next数组,要提前知道next数组的变化规律
next数组的变化规律:
1、若j的位置和i的位置匹配上了,则next[j + 1] = next[j] + 1,注意如果匹配上了,next数组一定是 + 1递增的,跨度只能是1
2、若j的位置和i的位置没匹配上,则j = next[j],回退到next[j]的位置
初始化:
求next数组部分:
1、该部分只涉及子串
2、next[1] = 0;
3、for遍历i,只走一次,不回退,且i初始化为2,因为next[1]已经被定义了
4、j从0开始
KMP部分:
1、for遍历i,只走一次,不回退,i初始化为1(注意这里和next数组的不同),且for用来遍历母串
2、j从0开始,遍历的是子串
AC代码:
const int N = 1e6 + 5;
int nextt[N];
void GetNext(const string& p)
{
nextt[1] = 0;
int n = p.size() - 1;//因为在字符串前面加了个'0',所以字符串的长度 = 字符串的size - 1
for (int i = 2, j = 0/**/; i <= n; i++)
{
while (j && p[j + 1] != p[i])
/*注意有一个 + 1,要判断的是下一个位置是否和i位置相等*/
j = nextt[j];//回退
if (p[j + 1] == p[i])//如果匹配上了
j++;//继续往后走,不回退
nextt[i] = j;//注意是next[i],并不是next[j]
}
}
void KMP(const string& s, const string& p)
{
int len1 = s.size() - 1;
int len2 = p.size() - 1;//因为在字符串前面加了个'0',所以字符串的长度 = 字符串的size - 1
for (int i = 1/**/, j = 0; i <= len1/*注意是遍历母串*/; i++)
{
while (j && p[j + 1] != s[i]/*子串的j + 1和母串的i不匹配*/)
j = nextt[j];//回退
if (p[j + 1] == s[i])//如果匹配上了
j++;//继续往后走,不回退
//题目要求的输出
if (j == len2)
cout << i - len2 + 1 << endl;
}
}
void solve()
{
string s, p;//母串,子串(模式串)
cin >> s >> p;
s = '0' + s;//注意下标从1开始
p = '0' + p;
GetNext(p);
KMP(s, p);
int len1 = s.size();
int len2 = p.size();
for (int i = 1/*从1开始*/; i < len2; i++)
cout << nextt[i] << " ";
cout << endl;
}
int main()
{
solve();
return 0;
}