[인프런 C++] 45. 공주 구하기 (조세퍼스)

🍧 문제

왕자가 N명이 있는데, 서로 공주를 구하러 가겠다고 한다. 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 한다. 왕은 왕자들을 나이 순으로 1~N번까지 차례로 번호를 매긴다. 그리고 1번 왕자부터 N번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다. 그리고 1번 왕자부터 시계방향으로 돌아가면 1부터 시작하여 번호를 외치게 한다. 한 왕자가 특정  숫자  K를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고, 원 밖으로 나오게 된다. 그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다. 이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.

예를 들어 8명의 왕자가 있을 때, 3을 외친 왕자가 제외된다고 하자. 처음에 3번 왕자가 3을 외쳐 제외되며, 이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7번 왕자가 공주를 구하러 간다.

N과 K가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.

 

✨ 1트 (강의 풀이)

1. 왕자 배열을 만들어서 모두 0으로 초기화

2. 값이 0이면서, k에 해당하면 +1 해준다.

3. position 값이 n 보다 크면 다시 1로 돌아온다.

4. break point가 n-1일 때, while문 탈출 및 배열의 값이 0인 왕자가 답이 된다.

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

int main(void){
	freopen("input.txt", "rt", stdin);
	
	int n, k, pos = 0, bp = 0, cnt = 0;
	scanf("%d %d", &n, &k);
	vector<int> prince(n+1);
	
	while(1){
		pos++;
		
		if(pos > n){
			pos = 1;
		}
		
		if(prince[pos] == 0){
			cnt++;
			if(cnt == k){
				prince[pos] = 1;
				cnt = 0;
				bp++;
			}
		}
		
		if(bp == n-1){
			break;
		}
	}
	
	for(int i = 1; i <= n; ++i){
		if(prince[i] == 0){
			printf("%d", i);
			break;
		}
	}
	
	return 0;
}