Python/코딩테스트

[백준 2178번] 미로 탐색

프로그린 2024. 4. 10. 22:34

문제상황

N×M크기의 배열로 표현되는 미로가 있다.

1 0 1 1 1 1

1 0 1 0 1 0

1 0 1 0 1 1

1 1 1 0 1 1

 

미로에서 1은 이동할 수 있는 칸을 나타내고, 0은 이동할 수 없는 칸을 나타낸다. 이러한 미로가 주어졌을 때, (1, 1)에서 출발하여 (N, M)의 위치로 이동할 때 지나야 하는 최소의 칸 수를 구하는 프로그램을 작성하시오. 한 칸에서 다른 칸으로 이동할 때, 서로 인접한 칸으로만 이동할 수 있다.

 

위의 예에서는 15칸을 지나야 (N, M)의 위치로 이동할 수 있다. 칸을 셀 때에는 시작 위치와 도착 위치도 포함한다.

 

입력조건

- 첫재 줄에 두 정수 N, M이 주어진다. (2 ≤ N, M ≤ 1000)
- 다음 N개의 줄에는 M개의 정수로 미로가 주어진다.
- 각각의 수들은 붙어서 입력으로 주어진다.
- 항상 도착 위치로 이동할 수 있는 경우에만 입력으로 주어진다.

 

출력조건

- 첫재 줄에 지나야 하는 최소의 칸 수를 출력한다.

 

입출력 예시

입출력 예 #1
입력 예시 : 4 6
                  101111
                  101010
                  101011
                  111011
출력 예시 : 15

입출력 예 #2
입력 예시 : 4 6
                  110110
                  110110
                  111111
                  111101
출력 예시 : 9

입출력 예 #3
입력 예시 : 2 25
                  1011101110111011101110111
                  1110111011101110111011101
출력 예시 : 38

 

아이디어

1. 입력으로 주어진 값을 적절한 자료형을 활용하여 그래프로 나타내기
2. BFS 알고리즘을 이용하여 L, R, U, D 중 이동할 수 있는 방향을 탐색
3. 1인 값을 찾으면 이동거리를 갱신해가며 (N, M) 위치까지 반복 작업

구현

n, m = map(int, input('미로의 크기를 입력하세요.').split(' '))
graph = [[] for _ in range(n)]
miro_data = [input("{}번째 미로를 입력하세요.".format(i+1)) for i in range(n)]
for row in range(n):
    for data in miro_data[row]:
        graph[row].append(int(data))
        
dx = [-1, 1, 0,  0]
dy = [ 0, 0, 1, -1]

from collections import deque
q = deque([(0, 0)])
while q:
    now_x, now_y = q.popleft()
    for i in range(4):
        next_x, next_y = now_x + dx[i], now_y + dy[i]
        if 0 <= next_x < n and 0 <= next_y < m:
            if graph[next_x][next_y] == 1:
                graph[next_x][next_y] = graph[now_x][now_y] + 1
                q.append((next_x, next_y))
graph[n-1][m-1]

 

문제 출처 : https://www.acmicpc.net/problem/2178

 

2178번: 미로 탐색

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

www.acmicpc.net