[프로그래머스] 게임 맵 최단거리(python)

문제 링크 - https://programmers.co.kr/learn/courses/30/lessons/1844


나의 풀이

from collections import deque
def solution(maps):
    
    def bfs(x, y):
        dx = [-1, 1, 0, 0]
        dy = [0, 0, -1, 1]
        
        queue = deque()
        queue.append((x, y))
        
        while queue:
            x, y = queue.popleft()
            
            for i in range(4):
                nx = x + dx[i]
                ny = y + dy[i]
                
                if nx < 0 or nx >=len(maps) or ny < 0 or ny >= len(maps[0]):
                    continue
                if maps[nx][ny] == 0:
                    continue
                    
                if maps[nx][ny] == 1:
                    maps[nx][ny] = maps[x][y] + 1
                    queue.append((nx, ny))
        return maps[len(maps)-1][len(maps[0])-1]
    if bfs(0, 0) == 1:
        return -1
    else:
        return bfs(0, 0)
  • 백준 미로탐색 문제와 비슷
  • bfs로 방문할 때마다 바로 전의 위치 값에 +1을 해주었다.
  • 그렇게 마지막 값을 return하면 답
  • 만약 마지막 값이 1이라면 벽에 막혀 상대방 진영까지 가지 못한 경우이므로 -1을 return


다른 사람의 풀이

from collections import deque

d = [[1,0], [-1, 0], [0,1], [0,-1]]

def solution(maps):
    r = len(maps)
    c = len(maps[0])
    table = [[-1 for _ in range(c)] for _ in range(r)]
    q = deque()
    q.append([0,0])
    table[0][0] = 1

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

        for i in range(4):
            ny = y + d[i][0]
            nx = x + d[i][1]

            if -1<ny<r and -1<nx<c:
                if maps[ny][nx] == 1:
                    if table[ny][nx] == -1:
                        table[ny][nx] = table[y][x] + 1
                        q.append([ny, nx])

    answer = table[-1][-1]
    return answer

좋은 웹페이지 즐겨찾기