【ybt高效进阶2-3-1】子串查找

子串查找

题目链接：ybt高效进阶2-3-1

题目大意

求一个字符串在另一个字符串的出现次数。

思路

这道题好像是 KMP 模板题。

就按着 KMP 的模板打一遍就可以了。

代码

#include<cstdio>
#include<cstring>

using namespace std;

int an, bn, ans, fail[1000001], j;
char a[1000001], b[1000001];

int main() {
	scanf("%s", a + 1);
	scanf("%s", b + 1);
	an = strlen(a + 1);
	bn = strlen(b + 1);
	
	j = 0;
	for (int i = 2; i <= bn; i++) {
		while (j && b[i] != b[j + 1]) j = fail[j];
		if (b[i] == b[j + 1]) j++;
		fail[i] = j;
	}
	
	j = 0;
	for (int i = 1; i <= an; i++) {
		while (j && a[i] != b[j + 1]) j = fail[j];
		if (a[i] == b[j + 1]) j++;
		if (j == bn) {
			ans++;
		}
	}
	
	printf("%d", ans);
	
	return 0;
}