-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathkmp.cpp
More file actions
56 lines (51 loc) · 1.11 KB
/
kmp.cpp
File metadata and controls
56 lines (51 loc) · 1.11 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include <iostream>
#include <vector>
#include <string>
using namespace std;
vector<int> getPi(string& p)
{
int n = (int)p.size(), j = 0;
vector<int> pi(n, 0);
for(int i = 1; i < n; i++)
{
while(j > 0 && p[i] != p[j])
j = pi[j-1];
if(p[i] == p[j])
pi[i] = ++j;
}
return pi;
}
vector<int> kmp(string& s, string& p)
{
vector<int> ans;
auto pi = getPi(p);
int n = (int)s.size(), m = (int)p.size(), j = 0;
for(int i = 0; i < n; i++)
{
while(j > 0 && s[i] != p[j])
j = pi[j-1];
if(s[i] == p[j])
{
if(j == m-1)
{
ans.push_back(i - m + 1);
j = pi[j];
}
else
{
j++;
}
}
}
return ans;
}
int main()
{
string s, p;
getline(cin, s); // 주어진 문자열
getline(cin, p); // 부분문자열
auto cnt = kmp(s, p);
cout << (int)cnt.size() << "\n"; //부분문자열의 개수
for(auto i : cnt) cout << i + 1 << " "; //부분문자열의 시작위치
return 0;
}