KMP题解代码(含讲解)

2024-05-26 08:56:57 浏览数 (1)

题目链接:【模板】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代码:

代码语言:javascript复制
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;
}

0 人点赞