KMP C++题解

KMP

内存限制： 128 MiB 时间限制： 1000 ms 标准输入输出 题目类型： 传统 评测方式： 文本比较

题目描述

给出两个字符串A,B。A的长度不超过1000000，B的长度不超过1000000。求B是否为A的子串，如果是，请输出B在A中第一次出现的位置。

输入格式

输入：
两行，第一行的字符串表示A,第二行的字符串表示B。

输出格式

输出：
如果B是A的子串，第一行输出“yes”，第二行输出B在A中第一次出现的位置。
否则只输出一行“no”。

样例

样例输入

复制ABBAABCD
BA

样例输出

复制yes
2

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;
int ans;
string a, b;
vector<int> nxt(N);
void get_next(string s) {
	int i = 0, j = -1, lens = s.size();
	nxt.resize(lens), nxt[0] = -1;
	while(i < lens) {
		if(j == -1 || s[i] == s[j]) {
			i++, j++;
			if(s[i] == s[j]) nxt[i] = nxt[j];
			else nxt[i] = j;
		} else j = nxt[j];
	} 
} 
void KMP(string f, string s) {
	int i = 0, j = 0, lenf = f.size(), lens = s.size();
	while(i < lenf) {
		if(j == -1 || f[i] == s[j]) {
			i++, j++;
			if(j == lens) {
				ans = i - j + 1;
				return;
			} 
		} else j = nxt[j];
	} 
} 
int main() {
	cin >> a >> b;
	get_next(b), KMP(a, b);
	if(!ans) printf("no");
	else printf("yes\n%d", ans - 1);
	return 0;
}