[백준/16928/파이썬] 뱀과 사다리 게임

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

16928번: 뱀과 사다리 게임

Problem

• 게임은 정육면체 주사위를 사용하며, 주사위의 각 면에는 1부터 6까지 수가 하나씩 적혀있다. 게임은 크기가 10×10이고, 총 100개의 칸으로 나누어져 있는 보드판에서 진행된다. 보드판에는 1부터 100까지 수가 하나씩 순서대로 적혀져 있다.
• 게임의 목표는 1번 칸에서 시작해서 100번 칸에 도착하는 것이다.
• 게임판의 상태가 주어졌을 때, 100번 칸에 도착하기 위해 주사위를 굴려야 하는 횟수의 최솟값을 구해보자.

조건

• 첫째 줄에 게임판에 있는 사다리의 수 N(1 ≤ N ≤ 15)과 뱀의 수 M(1 ≤ M ≤ 15)이 주어진다.
• 둘째 줄부터 N개의 줄에는 사다리의 정보를 의미하는 x, y (x < y)가 주어진다. x번 칸에 도착하면, y번 칸으로 이동한다는 의미이다.
• 다음 M개의 줄에는 뱀의 정보를 의미하는 u, v (u > v)가 주어진다. u번 칸에 도착하면, v번 칸으로 이동한다는 의미이다.
• 1번 칸과 100번 칸은 뱀과 사다리의 시작 또는 끝이 아니다. 모든 칸은 최대 하나의 사다리 또는 뱀을 가지고 있으며, 동시에 두 가지를 모두 가지고 있는 경우는 없다. 항상 100번 칸에 도착할 수 있는 입력만 주어진다.

SOL

 

처음에는 DFS로 풀려고 했지만, 최단 거리를 찾기에는 BFS가 더 적합할 것 같아서, BFS로 돌아줬다.

굳이 2차원 배열을 쓸 필요가 없어서, 1차원 배열로 돌아줬고, ladder , snake를 dictionary에 저장해줬다.

 

처음에는 1차원 배열에 idx = 번호, value = idx 이렇게 초기화 해놓고, x for x in range(101)

• if value >idx (사다리인 경우) : 방문한 적이 없다면, 해당 좌표로 이동. 
• if value < idx (뱀 인경우):  방문한 적이 없다면, 해당 좌표로 이동.

이렇게 구현하려고 했지만, 주사위를 굴려서 가야할 때의 처리가 구현이 어려워지는 부분이 생긴다. 그래서, dictionary에서 사다리를 따로 관리해주고, move 좌표를 한번 검사해준뒤, move 좌표가 사다리나 레더에 있으면, 그 좌표를 방문한적이 있는지 검사해주고 바뀐 곳에서 bfs를 진행해준다. (이때, 주사위를 굴리는 경우라면, 그냥 그대로 진행이 될거임!)

import sys
from collections import deque
input= sys.stdin.readline

dice=[1,2,3,4,5,6]
board=[0 for _ in range(101)]
visited=[False] * 101
N,M =map(int,input().split())

ladder=dict()
snake=dict()
for i in range(N):
    y,x=map(int,input().split())
    ladder[y]=x

for i in range(M):
    u,v=map(int,input().split())
    snake[u]=v

def game():
    dq=deque([1])
    visited[1] = True
    while dq:
        now =dq.popleft()
        for k in range(6):
            move=now+dice[k]
            if move<0 or move>100:
                continue
            if not visited[move]:
                if move in ladder.keys():
                    move = ladder[move]
                if move in snake.keys():
                    move= snake[move]
                # snake,ladder가 있으므로, 다시 그 좌표에 방문했는지, 안했는지 확인해야한다.
                if not visited[move]:
                    dq.append(move)
                    visited[move]=True
                    board[move] = board[now] +1

game()
print(board[100])