코딩테스트/백준

JAVA 백준 11729번 하노이 탑 이동 순서

광고(주) 2022. 6. 14. 16:22
반응형

문제

세 개의 장대가 있고 첫 번째 장대에는 반경이 서로 다른 n개의 원판이 쌓여 있다. 각 원판은 반경이 큰 순서대로 쌓여있다. 이제 수도승들이 다음 규칙에 따라 첫 번째 장대에서 세 번째 장대로 옮기려 한다.

  1. 한 번에 한 개의 원판만을 다른 탑으로 옮길 수 있다.
  2. 쌓아 놓은 원판은 항상 위의 것이 아래의 것보다 작아야 한다.
    이 작업을 수행하는데 필요한 이동 순서를 출력하는 프로그램을 작성하라. 단, 이동 횟수는 최소가 되어야 한다.
    아래 그림은 원판이 5개인 경우의 예시이다.

입력

첫째 줄에 첫 번째 장대에 쌓인 원판의 개수 N (1 ≤ N ≤ 20)이 주어진다.

3

출력

첫째 줄에 옮긴 횟수 K를 출력한다.
두 번째 줄부터 수행 과정을 출력한다. 두 번째 줄부터 K개의 줄에 걸쳐 두 정수 A B를 빈칸을 사이에 두고 출력하는데, 이는 A번째 탑의 가장 위에 있는 원판을 B번째 탑의 가장 위로 옮긴다는 뜻이다.

7
1 3
1 2
3 2
1 3
2 1
2 3
1 3

알고리즘 분류

재귀(recursion)

소스코드

import java.io.*;
import java.util.Arrays;

public class Main {
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    static Top[] top = new Top[3];

    public static void hanoit(boolean odd, int a, int b) throws IOException {
        if (top[a].isEmpty()) {
            return;
        }
        int p = top[a].pop();
        top[b].push(p);
        bw.write((a + 1) + " " + (b + 1) + "\n");
        if (odd) {
            a = ++a % 3;
            b = ++b % 3;
        } else {
            a = (a + 2) % 3;
            b = (b + 2) % 3;
        }

        if (top[a].isEmpty() || (!top[b].isEmpty() && top[a].get() > top[b].get())) {
            int temp = a;
            a = b;
            b = temp;
        }
        hanoit(odd, a, b);
    }

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        for (int i = 0; i < 3; i++) {
            top[i] = new Top(N);
        }
        for (int i = N; i > 0; i--) {
            top[0].push(i);
        }
        bw.write(((int) Math.pow(2, N)) - 1 + "\n");
        hanoit(N % 2 == 1, 0, N % 2 + 1);
        bw.flush();
    }

    static class Top {
        int[] top;
        int index;

        Top(int size) {
            top = new int[size];
            index = size;
        }

        boolean isEmpty() {
            return index == top.length;
        }

        int get() {
            if (isEmpty()) return 0;
            return top[index];
        }

        int pop() {
            int p = top[index];
            top[index++] = 0;
            return p;
        }

        void push(int val) {
            top[--index] = val;
        }

        String print() {
            return Arrays.toString(top);
        }
    }
}
반응형