문제
정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다.
정보 왕국에는 왕자가 N명이 있는데 서로 공주를 구하러 가겠다고 합니다. 정보왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했습니다.
왕은 왕자들을 나이 순으로 1번부터 N번까지 차례로 번호를 매긴다. 그리고 1번 왕자부터 N 번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다. 그리고 1번 왕자부터 시 계방향으로 돌아가며 1부터 시작하여 번호를 외치게 한다. 한 왕자가 K(특정숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 된다. 그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다.
이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.
예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자. 처음에는 3번 왕자가 3 을 외쳐 제외된다. 이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7 번 왕자에게 공주를 구하러갑니다. N과 K가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.
💡 입력설명
첫 줄에 자연수 N(5<=N<=1,000)과 K(2<=K<=9)가 주어진다.
💡 출력설명
첫 줄에 마지막 남은 왕자의 번호를 출력합니다.
💡 입력예제
8 3
💡 출력예제
7
코드
최종 풀이
#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
int n, k, i, pos = 0, cnt = 0, bp = 0;
scanf("%d %d", &n, &k);
vector<int> a(n + 1);
while(1) {
pos++;
if(pos > n) {
pos -= n;
}
if(a[pos] == 0) {
cnt++;
if(cnt == k) {
a[pos]++;
cnt = 0;
bp++;
}
}
if(bp == n - 1) {
break;
}
}
for(i = 1; i <= n; i++) {
if(a[i] == 0) {
printf("%d", i);
}
}
return 0;
}
설명
1~N명의 왕자로 구성하기 위해 index를 1로 시작하고 0으로 초기화된 vector a를 생성한다.
각 index를 지칭하는 pos와 k번째에 도달하는지를 판단하는 cnt, 무한반복문에서 n-1까지 입력받았을 경우 break할 bp를 선언한다.
무한 반복문을 작성하여 pos를 1씩 증가하도록 한다. pos에 위치하는 a의 값이 0일 경우 cnt를 1씩 증가하고 해당 cnt가 사용자로부터 입력받은 특정숫자 k와 일치하면 pos에 위치하는 a의 값을 1증가하고 bp또한 1증가시킨다. 이후 다음 k번째의 index에 판단하기 위해 cnt를 0으로 초기화한다.
위를 반복하여 bp가 n-1에 도달했을 경우 break하여 무한 반복문에 빠져나온 후 해당 배열에서의 값이 0인 index를 찾아 출력한다.
해당 내용은 김태원님의 'it 취업을 위한 알고리즘 문제풀이' 강의를 듣고 작성한 글입니다.
