[1786번] 찾기 (KMP Algorithm)

문제 출처 :

https://www.acmicpc.net/problem/1786

알고리즘 분석 :

문제 해결에 필요한 사항

1. KMP Algorithm

문자열 찾기가 조금 복잡해진다 싶으면 KMP 알고리즘이 최적 해이다.

O(N+M)이기에 엄청난 효율성이 있지만, 코드 및 이해가 어렵다는 단점이 있다.

KMP 알고리즘의 존재를 알고는 있지만 아직 완벽하지 않기에 

기술하기가 어려워 다른분의 블로그 내용을 가져오고자 한다.

KMP 알고리즘에 대해 더 공부하고 나만의 해석으로 알고리즘 게시판에 향후 올리려 한다.

소스코드에 대한 원문 및 KMP 알고리즘 설명은

http://bowbowbow.tistory.com/6 에 있습니다.

소스 코드 : 

#include <iostream>
#include <string>
#include <cstdio>
#include <vector>
 
using namespace std;
 
vector<int> getPi(string p) 
{
    int m = (int)p.size();
    int j = 0;
 
    vector<int> pi(m, 0);
 
    for (int i = 1; i< m; 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 matched = kmp(s, p);
 
    printf("%d\n", (int)matched.size());
 
    for (auto i : matched)
        printf("%d ", i + 1);
 
    return 0;
}