글 읽기 - 파이썬 구슬탈출, return 코드를 넣었더니 틀렸다고 하네요

sososunny0317 4년 전 0

if dist[(rx, ry, bx, by)] >= 10:
print(0)
return

움직인 횟수가 10번 넘어가면 프로그램을 종료하려고 return을 하였습니다.

하지만 이렇게 제출하면 틀렸다고 나오네요ㅠㅠ

if dist[(rx, ry, bx, by)] >= 10:
break
.

print(0)

으로 고치면 맞았다고 나오구요

return 코드로 할 때 틀리는 이유가 뭘까요ㅠㅠ?

from collections import deque

n, m = map(int, input().split())
a = [list(map(str, input())) for _ in range(n)]
dist = {}
dx, dy = [-1, 1, 0, 0], [0, 0, -1, 1]

# 처음 구슬의 위치
q = deque()
for i in range(n):
    for j in range(m):
        if a[i][j] == 'R':
            rx, ry = i, j
        elif a[i][j] == 'B':
            bx, by = i, j
q.append((rx, ry, bx, by))
dist[(rx, ry, bx, by)] = 0


# 한 방향으로 쭈우우우욱 이동
def move(_x, _y, _dx, _dy, _c):
    while a[_x + _dx][_y + _dy] != '#' and a[_x][_y] != 'O':
        _x += _dx
        _y += _dy
        _c += 1
    return _x, _y, _c

def bfs():
    while q:
        rx, ry, bx, by = q.popleft()

        if dist[(rx, ry, bx, by)] >= 10:
            print(0)
            return

        for k in range(4):
            nrx, nry, rc = move(rx, ry, dx[k], dy[k], 0)
            nbx, nby, bc = move(bx, by, dx[k], dy[k], 0)

            if a[nbx][nby] == 'O':
                continue
            if a[nrx][nry] == 'O':
                print(1)
                return

            if nrx == nbx and nry == nby:
                if rc > bc:
                    nrx, nry = nrx - dx[k], nry - dy[k]
                else:
                    nbx, nby = nbx - dx[k], nby - dy[k]

            if not (nrx, nry, nbx, nby) in dist.keys():
                q.append((nrx, nry, nbx, nby))
                dist[(nrx, nry, nbx, nby)] = dist[(rx, ry, bx, by)] + 1

bfs()

djm03178 4년 전 1

10번을 움직이기 전에 모든 상태를 방문해서 큐가 비게 될 때 아무것도 출력하지 않습니다.

3 6
######
#RB#O#
######

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

문제

문제

출처

ICPC

파이썬 구슬탈출, return 코드를 넣었더니 틀렸다고 하네요

sososunny0317 4년 전 0

djm03178 4년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말