알고리즘 정복하기!/백준 문제풀이

백준 2178번 Python / BFS

by seokii 2022. 1. 25.
728x90
반응형

너비 우선 탐색(BFS)이란?

https://seokii.tistory.com/31

 

[알고리즘] 너비 우선 탐색 (BFS, Breadth-First Search)이란?

1. 너비 우선 탐색 (BFS, Breadth-First Search) - 그래프에서 가까운 노드부터 우선적으로 탐색하는 알고리즘 - 큐 자료구조를 이용함 장점 - 출발노드에서 목표노드까지의 최단 길이 경로를 보장한다.

seokii.tistory.com

너비 우선 탐색 알고리즘 개념에 대한 내용입니다.

 

문제링크

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

 

2178번: 미로 탐색

첫째 줄에 두 정수 N, M(2 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

www.acmicpc.net

 

풀이

from collections import deque

n, m = map(int, input().split())
graph = []

for _ in range(n):
    graph.append(list(map(int, input())))

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>=n or ny<0 or ny>=m:
                continue
            
            if graph[nx][ny] == 0:
                continue
            
            if graph[nx][ny] == 1:
                graph[nx][ny] = graph[x][y] + 1
                queue.append((nx, ny))

    return graph[n-1][m-1]

print(bfs(0,0))
728x90
반응형

댓글