설명

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다.

정보 왕국에는 왕자가 N명이 있는데 서로 공주를 구하러 가겠다고 합니다.

정보왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했습니다.

왕은 왕자들을 나이 순으로 1번부터 N번까지 차례로 번호를 매긴다.

그리고 1번 왕자부터 N번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다.

그리고 1번 왕자부터 시계방향으로 돌아가며 1부터 시작하여 번호를 외치게 한다.

한 왕자가 K(특정숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 된다.

그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다.

이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.

Image1.jpg

예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자. 처음에는 3번 왕자가 3을 외쳐 제외된다.

이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7번 왕자에게 공주를 구하러갑니다.

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

코드

public class SavePrincess {
    public static void main(String[] args){
        Scanner in=new Scanner(System.in);
        int input1 = in.nextInt();
        int input2 = in.nextInt();

        int solution = solution(input1, input2);
        System.out.println(solution);
    }
    public static int solution(int su1, int su2){
        Queue<Integer> que = new LinkedList<>();
        for(int i=1; i<=su1; i++){
            que.add(i);
        }

        int cnt = 0;
        while(que.size() != 1){
            cnt ++;
            int su = que.poll();
            if(cnt != su2)
                que.offer(su);
            else
                cnt = 0;
        }

        return que.poll();
    }
}

Queue 기본 문제이다. Queue 관련 문제는 많이 풀어보지 않아서 처음에 poll()해버린 값을 다시 que에 넣어야한다는 생각을 하지 못하고 분명 que는 원처럼 계속 값을 가지고 있어야하는데... 어떻게 하더라... 하다가 뺀 걸 다시 넣으면 된다는걸 깨달았다... 바본가...ㅎ 그래서 결국 문제는 풀었고 queue도 stack 만큼 알면 재밌는 자료구조인거 같다. 앞으로 순환하는 문제는 queue로 풀어봐야겠다.

queue는 넣을때 offer(), 뺄때 poll()