정보
문제 바로가기 [클릭]
난이도: Silver1
관련 개념: #그래프 이론 #그래프 탐색 #너비 우선 탐색
조건
시간 제한 | 메모리 제한 |
---|---|
1 초 | 192 MB |
문제
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 ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.
출력
첫째 줄에 지나야 하는 최소의 칸 수를 출력한다. 항상 도착위치로 이동할 수 있는 경우만 입력으로 주어진다.
예제 입출력 1
입력
1
2
3
4
5
4 6
101111
101010
101011
111011
출력
1
15
예제 입출력 2
입력
1
2
3
4
5
4 6
110110
110110
111111
111101
출력
1
9
예제 입출력 3
입력
1
2
3
2 25
1011101110111011101110111
1110111011101110111011101
출력
1
38
예제 입출력 4
입력
1
2
3
4
5
6
7
8
7 7
1011111
1110001
1000001
1000001
1000001
1000001
1111111
출력
1
13
코드(파이썬)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import sys
def bfs(start):
cnt = 1
queue = [start]
visited = set()
direction = [(1, 0), (-1, 0), (0, 1), (0, -1)]
while (n-1, m-1) not in queue:
cnt += 1
for _ in range(len(queue)):
s = queue.pop(0)
for d in direction:
e = (s[0]+d[0], s[1]+d[1])
if e[0] > -1 and e[0] < n and\
e[1] > -1 and e[1] < m and\
maze[e[0]][e[1]] and e not in visited:
visited.add(e)
queue.append(e)
return cnt
n, m = map(int, sys.stdin.readline().split())
maze = [tuple(map(int, list(l.rstrip("\n")))) for l in sys.stdin.readlines()]
print(bfs((0, 0)))
특이사항
- DFS와 BFS를 동시에 사용
- 내 코드(31,376KB, 88ms)에 비해 우수한 코드 (바로가기, 30,864KB, 88ms)와 비교
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
import sys n, m = map(int, sys.stdin.readline().split()) arr = [list(map(int, sys.stdin.readline().rstrip())) for _ in range(n)] dx = [1, -1, 0, 0] dy = [0, 0, -1, 1] queue = [[0, 0]] while queue: a, b = queue[0][0], queue[0][1] del queue[0] for i in range(4): x = a + dx[i] y = b + dy[i] if 0 <= x < n and 0 <= y < m and arr[x][y] == 1: queue.append([x, y]) arr[x][y] = arr[a][b] + 1 print(arr[n - 1][m - 1])
- 방문 여부
- 기존: visited set으로 방문 여부 확인
- 개선: queue 하나로 해결
- maze, arr 배열(입력 배열)
- 기존: maze에 수정 없음
- 개선: 각 위치까지의 최단 경로
참고문헌
-
Comments powered by Disqus.