[백준] 11866번 요세푸스 문제 0

www.acmicpc.net/problem/11866

11866번: 요세푸스 문제 0

벡터의 원소를 직접 제거해가면서 풀었다.

주의할 점은 벡터의 크기가 점점 줄어들기 때문에 index가 벡터의 범위를 넘어가지 않게 % 연산자를 이용해 조정해줘야 한다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
	int n, k;
	cin >> n >> k;
	vector<int> v(n);
	for (int i = 0; i < n; i++) {
		v[i] = i + 1;
	}
	int index = 0;
	while (!v.empty()) {
		if (v.size() == n) {
			cout << '<';
		}
		index += (k - 1);
		index %= v.size();
		cout << v[index];
		v.erase(v.begin() + index);
		if (v.empty()) {
			cout << '>';
		}
		else {
			cout << ", ";
		}
	}
	return 0;
}