백준 1158 _ 요세푸스 문제

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

1158번: 요세푸스 문제

---

⭐ 내 소스 코드

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


int main(void) {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	int n, k,index=0;
	vector<int>vec;

	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		vec.push_back(i);
	}
	cout << "<";
	while (1)
	{
		if (vec.size() == 1) {
			cout << vec[0];
			cout << ">";
			break;
		}
		else {
			index = (index + (k - 1)) % vec.size(); //k번째 인덱스 계산
			cout << vec[index] << ", "; 
			vec.erase(vec.begin() + index); //해당 인덱스 요소 제거
		}
		
	}

	return 0;
}

 

🙏 문제 풀이

index = index + (k - 1)) % vec.size() : 현재 인덱스에서 k만큼의 인덱스를 더하고, 인덱스는 0부터 시작이니 -1를 한 뒤 계산한 인덱스가 벡터의 사이즈보다 클 수도 있으니 현재 벡터의 사이즈(vec.size())로 나눈 몫을 인덱스로 사용한다.

 

vec.arease(vec.begin() + index) : 벡터에서 특정 위치의 요소 제거