글 읽기 - 이동한 칸의 개수 & 방문체크용으로 Python dictionary를 사용했는데요

sososunny0317 4년 전 0

원래 dist를 3차원 배열로 사용했었는데

dictionary를 사용해서 풀어보려고 합니다.

근데 문제 테스트 케이스는 답을 맞췄는데

런타임 에러가 나네요 ㅠ

어느 부분을 놓친걸까요?

도움 부탁드립니다~!

from collections import deque

n, m = map(int, input().split())
a = [list(map(int, input())) for _ in range(n)]

dist = {}
dx, dy = [-1, 1, 0, 0], [0, 0, -1, 1]

# 시작점
q = deque()
# x, y 좌표 / z 벽을 부순 개수
q.append((0, 0, 0))
dist[(0, 0, 0)] = 1

while q:
    x, y, z = q.popleft()

    for k in range(4):
        nx, ny = x + dx[k], y + dy[k]

        if 0 <= nx < n and 0 <= ny < m:
            # 벽을 부수지 않고 지나가는 경우
            if a[nx][ny] == 0 and not (nx, ny, z) in dist.keys():
                q.append((nx, ny, z))
                dist[(nx, ny, z)] = dist[(x, y, z)] + 1

            # 벽을 부수고 지나가는 경우
            if z == 0 and a[nx][ny] == 1 and not (nx, ny, z + 1) in dist.keys():
                q.append((nx, ny, z + 1))
                dist[(nx, ny, z + 1)] = dist[(x, y, z)] + 1

# 출력

# and 로 (n-1, m-1, 0), (n-1, m-1, 1) 둘 다 포함되어 있는지 확인할려니까 하나만 true면 true로 넘어감. 왜그러지?
if (n - 1, m - 1, 0) in dist.keys():
    if (n-1, m-1, 1) in dist.keys():
        print(min(dist[(n - 1, m - 1, 0), (n - 1, m - 1, 1)]))
elif (n - 1, m - 1, 0) in dist.keys():
    print(dist[(n - 1, m - 1, 0)])
elif (n-1, m-1, 1) in dist.keys():
    print(dist[(n - 1, m - 1, 1)])
else:
    print(-1)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

이동한 칸의 개수 & 방문체크용으로 Python dictionary를 사용했는데요

sososunny0317 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말