글 읽기 - 반복문 내 첨자 질문드립니다.

poww56 1년 전 0

두 함수 비슷하게 쭉 쓰고 1부터 실행하면 bfs1은 실행이 안되고 2번은 되네요.

함수 작성 시 어떤 부분을 조심하면서 작성해야 되는 문제인가요?

from collections import deque
n, m = map(int, input().split())
maze = [list(map(int, input())) for _ in range(n)]
# print(maze)


def bfs1(a, b):
    dx = [1, -1, 0, 0]
    dy = [0, 0, 1, -1]
    q = deque()
    q.append((a, b))
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if not(0 <= nx < n and 0 <= ny < m):
                continue
            if maze[nx][ny] == 0:
                continue
            if maze[nx][ny] == 1:
                maze[nx][nx] = maze[x][y] + 1
                q.append((nx, ny))
    return maze[n - 1][m - 1]


def bfs2(x, y):
    dx = [1, -1, 0, 0]
    dy = [0, 0, 1, -1]
    q = deque()
    q.append((x, y))
    while q:
        a, b = q.popleft()
        for i in range(4):
            na = a + dx[i]
            nb = b + dy[i]
            if not (0 <= na < n and 0 <= nb < m):
                continue
            if maze[na][nb] == 0:
                continue
            if maze[na][nb] == 1:
                maze[na][nb] = maze[a][b] + 1
                q.append((na, nb))
    return maze[n - 1][m - 1]


print(bfs1(0, 0))
print(bfs2(0, 0))

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

문제

문제

출처

ICPC

반복문 내 첨자 질문드립니다.

poww56 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말