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