BOJ: 1158 요세푸스 문제

문제

요세푸스 문제는 다음과 같다.

\(1\)번부터 \(N\)번까지 \(N\)명의 사람이 원을 이루면서 앉아있고, 양의 정수 \(K(\leq N)\)가 주어진다. 이제 순서대로 \(K\)번째 사람을 제거한다. 한 사람이 제거되면 남은 사람들로 이루어진 원을 따라 이 과정을 계속해 나간다. 이 과정은 \(N\)명의 사람이 모두 제거될 때까지 계속된다. 원에서 사람들이 제거되는 순서를 \((N, K)\)-요세푸스 순열이라고 한다. 예를 들어 \((7, 3)\)-요세푸스 순열은 \(<3, 6, 2, 7, 5, 1, 4>\)이다.

\(N\)과 \(K\)가 주어지면 \((N, K)\)-요세푸스 순열을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 \(N\)과 \(K\)가 빈 칸을 사이에 두고 순서대로 주어진다. \((1 \leq K \leq N \leq 5,000)\)

출력

예제와 같이 요세푸스 순열을 출력한다.

예제 입력 1

7 3

예제 출력 1

<3, 6, 2, 7, 5, 1, 4>

코드

#include <bits/stdc++.h>
using namespace std;
int main(void) {
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    vector<int> v;
    int n, k, c = 0;
    cin>>n>>k;
    for(int i=0; i<n; i++)
        v.push_back(i+1);
    cout<<'<';
    while(v.size() != 1) {
        
        for(int i=0; i<k; i++) {
            if(c >= n)
                c = c%n;
            c += 1;
        }
        
        cout<<v[c-1]<<','<<' ';
        
        v.erase(v.begin()+c-1);
        c--;
        n--;
    }
    cout<<v[0]<<'>';
    return 0;
}

Reference

BOJ