[BOJ] 2178. 미로 탐색
미로 탐색 성공분류
| 시간 제한 | 메모리 제한 | 제출 | 정답 | 맞은 사람 | 정답 비율 |
|---|---|---|---|---|---|
| 1 초 | 192 MB | 70115 | 26151 | 16636 | 36.152% |
문제
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
4 6 101111 101010 101011 111011
예제 출력 1
15
예제 입력 2
4 6 110110 110110 111111 111101
예제 출력 2
9
예제 입력 3
2 25 1011101110111011101110111 1110111011101110111011101
예제 출력 3
38
예제 입력 4
7 7 1011111 1110001 1000001 1000001 1000001 1000001 1111111
예제 출력 4
13
풀이
처음에는 DFS 라고 생각했다. (왜그랬지???) 한 쪽으로 쭉 가다가 목적지에 도착하면 그 count 값을 리스트에 넣고 마지막엔 그 리스트의 최소값을 출력하는 코드를 작성해서 제출했지만 시간 초과가 떴다. 재귀 DFS는 visited 인자를 가지고 새로운 인덱스 i,j가 visited에 포함되어있는지 'in' 으로 확인했는데, 매번 선형 탐색을 해야 하기 때문에 시간초과가 나서 BFS로 풀어야 한다. BFS로 풀면 최소값을 찾은 순간 return을 시켜 프로그램을 중지 시킬 수 있으므로 괜히 더 큰 값을 비교해야 하는 DFS보다 이 문제에서 더 적합하다고 결론을 내렸다.
또한 visited는 in 을 사용하지 않기 위해 2차원 bool 리스트로 선언하여 사용했다. 큐의 가장 앞에 있는 인덱스에 상하좌우 네가지 방향으로
1. 인덱스가 리스트 크기 범위 안에 있는지 체크, (이과정을 먼저 거치지 않으면 런타임오류가 발생할것)
2. 새로운 인덱스가 1 인지(길인지 체크)
3. 새로운 인덱스가 visited 되었는지 체크
모든 체크를 통과하면 큐에 새 인덱스를 추가한다.
큐에서 pop한 현재 인덱스가 목적지일 경우 프로그램을 종료한다.
코드
from collections import deque import sys n,m = map(int,input().split()) data = [] for _ in range(n): data.append(list(map(int,list(sys.stdin.readline().rstrip())))) goal = (n-1,m-1) def maze(i,j): q = deque([(0,0)]) direction = [[0,1],[1,0],[-1,0],[0,-1]] visited = [[False]*m for _ in range(n)] visited[0][0]=True count = 0 while q: count+=1 len_q = len(q) for temp in range(len_q): current = q.popleft() i = current[0] j = current[1] if (i,j) == goal: return count for d in direction: ni = i+d[0] nj = j+d[1] if 0<=ni<n and 0<=nj<m and data[ni][nj]==1 and visited[ni][nj]==False: q.append((ni,nj)) visited[ni][nj] = True print(maze(0,0)) | cs |
No comments: