[백준] 16928. 뱀과 사다리 게임/Python - Gold5

❓문제

https://www.acmicpc.net/problem/16928

성능 요약

메모리: 34968 KB, 시간: 60 ms

분류

너비 우선 탐색, 그래프 이론, 그래프 탐색

 

문제 설명

뱀과 사다리 게임을 즐겨 하는 큐브러버는 어느 날 궁금한 점이 생겼다.

주사위를 조작해 내가 원하는 수가 나오게 만들 수 있다면, 최소 몇 번만에 도착점에 도착할 수 있을까?

게임은 정육면체 주사위를 사용하며, 주사위의 각 면에는 1부터 6까지 수가 하나씩 적혀있다. 게임은 크기가 10×10이고, 총 100개의 칸으로 나누어져 있는 보드판에서 진행된다. 보드판에는 1부터 100까지 수가 하나씩 순서대로 적혀져 있다.

플레이어는 주사위를 굴려 나온 수만큼 이동해야 한다. 예를 들어, 플레이어가 i번 칸에 있고, 주사위를 굴려 나온 수가 4라면, i+4번 칸으로 이동해야 한다. 만약 주사위를 굴린 결과가 100번 칸을 넘어간다면 이동할 수 없다. 도착한 칸이 사다리면, 사다리를 타고 위로 올라간다. 뱀이 있는 칸에 도착하면, 뱀을 따라서 내려가게 된다. 즉, 사다리를 이용해 이동한 칸의 번호는 원래 있던 칸의 번호보다 크고, 뱀을 이용해 이동한 칸의 번호는 원래 있던 칸의 번호보다 작아진다.

게임의 목표는 1번 칸에서 시작해서 100번 칸에 도착하는 것이다.

게임판의 상태가 주어졌을 때, 100번 칸에 도착하기 위해 주사위를 굴려야 하는 횟수의 최솟값을 구해보자.

✍🏻풀이

이 문제는 BFS로 풀면 된다.

사다리와 뱀을 사전형으로 저장한다. 사전형으로 저장하는 편이 메모리를 줄일 수 있다.

출발지가 1이라고 하였으므로 큐에 1을 넣어주고 그 뒤에는 주사위를 굴리면 된다. cnt 값을 같이 큐에 넣을 수 있지만 이렇게 하면 메모리가 증가하게 된다. 그러면 메모리 초과로 실패한다. 그러므로 우리는 while q 안에 q 길이 만큼 반복한 후 cnt + 1을 한다.
1일 때 우리가 갈 수 있는 위치는 2, 3, 4, 5, 6, 7 이다. 주사위를 굴린 횟수는 어느 위치든 결국 1이다. 그러므로 이를 큐에 저장해서 메모리를 더 사용하기 보다. 반복문을 벗어났을 때 1 증가하는 편이 더 낫다.

위 방식을 반복하여 100에 도착하면 그 때의 cnt를 출력한 후 함수를 반환하면서 끝낸다. 

💻코드

처음 시도한 코드

import sys
from collections import deque

input = sys.stdin.readline

N, M = map(int, input().split()) # N -> 사다리 수, M -> 뱀 수
ladders = [list(map(int, input().split())) for _ in range(N)]
snakes = [list(map(int, input().split())) for _ in range(M)]

def bfs():
    global res
    q = deque()
    q.append((1, 0))

    while q:
        cur, cnt = q.popleft()
        for i in range(N):
            if ladders[i][0] == cur:
                cur = ladders[i][1]

        for i in range(M):
            if snakes[i][0] == cur:
                cur = snakes[i][1]

        for i in range(1, 7):
            nx = cur + i
            if nx == 100:
                res = cnt + 1
                return

            if nx < 100:
                q.append((nx, cnt + 1))
                res = cnt + 1


res = 0
bfs()
print(res)

➡️ 메모리 초과로 실패

정답 코드

import sys
from collections import deque

input = sys.stdin.readline

N, M = map(int, input().split())
ladders = {}
for _ in range(N):
    x, y = map(int, input().split())
    ladders[x] = y
snakes = {}
for _ in range(M):
    x, y = map(int, input().split())
    snakes[x] = y

def dfs():
    visited = [0] * 101
    visited[1] = 1
    q = deque()
    q.append(1)
    cnt = 0
    while q:
        l = len(q)
        for _ in range(l):
            cur = q.popleft()
            if cur == 100:
                print(cnt)
                return

            for i in range(1, 7):
                nx = cur + i
                if nx <= 100 and not visited[nx]:
                    visited[nx] = 1
                    if nx in ladders.keys():
                        visited[ladders[nx]] = 1
                        q.append(ladders[nx])
                    elif nx in snakes.keys():
                        visited[snakes[nx]] = 1
                        q.append(snakes[nx])
                    else:
                        q.append(nx)

        cnt += 1

dfs()

📍후기

뭔가 그냥 리스트로 푸는 게 당연하다고 생각했는데 앞으로 python으로 풀 때 메모리 줄이는 방법에 대해서도 충분히 고민을 해봐야겠다.