본문 바로가기

Computer Science/알고리즘

[Programmers] Level3 - 하노이의 탑 (연습문제) Python 풀이

728x90
반응형

문제

하노이 탑(Tower of Hanoi)은 퍼즐의 일종입니다. 세 개의 기둥과 이 기동에 꽂을 수 있는 크기가 다양한 원판들이 있고, 퍼즐을 시작하기 전에는 한 기둥에 원판들이 작은 것이 위에 있도록 순서대로 쌓여 있습니다. 게임의 목적은 다음 두 가지 조건을 만족시키면서, 한 기둥에 꽂힌 원판들을 그 순서 그대로 다른 기둥으로 옮겨서 다시 쌓는 것입니다.

한 번에 하나의 원판만 옮길 수 있습니다.
큰 원판이 작은 원판 위에 있어서는 안됩니다.
하노이 탑의 세 개의 기둥을 왼쪽 부터 1번, 2번, 3번이라고 하겠습니다. 1번에는 n개의 원판이 있고 이 n개의 원판을 3번 원판으로 최소 횟수로 옮기려고 합니다.

1번 기둥에 있는 원판의 개수 n이 매개변수로 주어질 때, n개의 원판을 3번 원판으로 최소로 옮기는 방법을 return하는 solution를 완성해주세요.

제한사항

  • n은 15이하의 자연수 입니다.

입출력 예

n result
2 [ [1,2], [1,3], [2,3] ]

풀이

이 문제는 하노이 탑에 대한 문제이다.
하노이 탑은 재귀를 설명할 때 예시를 들어 설명하기 좋은 예제이다.
이 문제를 풀이할 때는 재귀를 사용하기 위해 hanoi함수를 구현하였고, 값을 반환하기 위해 전역 변수로 값을 선언하고 각 함수에서 수정이 가능하도록 global로 각 함수에서 선언을 하여 풀이를 진행하였다.

이 문제는 Level 3에 있지만 재귀를 잘 알고 있으면 쉽게 풀이를 할 수 있을 정도이다.

코드

# 재귀를 사용
# 전역변수를 사용하고 global을 사용하여 각 함수에서도 값을 바꿀 수 있도록 선언
result = []

def hanoi(n, start, end, mid):
    global result

    if n == 1:
        result.append([start, end])
    else:
        hanoi(n-1 , start, mid, end)
        result.append([start, end])
        hanoi(n-1, mid, end, start)

    return result

def solution(n):
    global result

    hanoi(n, 1, 3, 2)

    return result
728x90
반응형