BOJ 1498 : 주기문

• 문제 링크 : boj.kr/1498
• 난이도 : P4
• 태그 : KMP

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

vector<int> getPi(string P) {
    int L = (int)P.size();
    vector<int> Pi(L,0);
    int j = 0;
    for(int i = 1; i < L; ++i) {
        while(j > 0 && P[i] != P[j])
            j = Pi[j-1];
        if(P[i] == P[j])
            Pi[i] = ++j;
    }
    return Pi;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    string s;
    cin >> s;

    auto Pi = getPi(s);
    for(int i = 0; i < s.length(); i++) {
        int len = i+1 - Pi[i];
        if(Pi[i] != 0 && (i+1) % len == 0)
            cout << (i+1) << " " << (i+1)/len << "\n";
    }

    return 0;
}

---

풀이

#1305를 응용하여 풀 수 있다.

문자열의 길이가 (문자열의 길이 - 문자열의 마지막 pi값)으로 나누어떨어지면 주기문이 된다.