0
1
mirror of https://git.sb/baoshuo/OI-codes.git synced 2024-09-16 20:25:24 +00:00

P3375 【模板】KMP字符串匹配

https://www.luogu.com.cn/record/91905614
This commit is contained in:
Baoshuo Ren 2022-10-28 17:27:23 +08:00
parent d7c1104f8b
commit 17e280db2a
Signed by: baoshuo
GPG Key ID: 00CB9680AB29F51A

View File

@ -3,32 +3,41 @@
using std::cin;
using std::cout;
using std::endl;
const char endl = '\n';
const int N = 1000005;
const int N = 1e6 + 5;
int f[N], next[N];
int next[N], f[N];
std::string s1, s2;
int main() {
std::ios::sync_with_stdio(false);
cin.tie(nullptr);
cin >> s1 >> s2;
next[0] = -1;
for (int i = 1, j = -1; i < s2.size(); i++) {
while (j != -1 && s2[i] != s2[j + 1]) j = next[j];
while (~j && s2[i] != s2[j + 1]) j = next[j];
if (s2[i] == s2[j + 1]) j++;
next[i] = j;
}
for (int i = 0, j = -1; i < s1.size(); i++) {
while (j != -1 && (j == s2.size() || s1[i] != s2[j + 1])) j = next[j];
while (~j && (j == s2.size() || s1[i] != s2[j + 1])) j = next[j];
if (s1[i] == s2[j + 1]) j++;
if (j == s2.size() - 1) {
if (j + 1 == s2.size()) {
cout << i - j + 1 << endl;
f[i] = j;
}
}
for (int i = 0; i < s2.size(); i++) {
cout << next[i] + 1 << ' ';
}
cout << endl;
return 0;
}