coding test

[파이썬] 2178. 미로 탐색

잔망루피 2021. 5. 22. 22:33

문제

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

 

 

👧 나의 풀이

 

import sys
from collections import deque

N, M = map(int, sys.stdin.readline().split())
maze = list()
visited = [[0] * M for _ in range(N)]

for i in range(N):
    maze.append(sys.stdin.readline().rstrip())

def check(x, y):
    return False if x >= N or y >= M or x < 0 or y < 0 else True

def bfs():
    que=deque()
    que.append((0, 0))
    visited[0][0]=1

    while que :
        x, y=que.popleft()
        if x == N-1 and y == M-1 :
            return visited[x][y]
        for i, j in [(0, 1), (0, -1), (-1, 0), (1, 0)] :
            if check(x+i, y+j) :
                if not visited[x+i][y+j] and maze[x+i][y+j] == '1':
                    visited[x+i][y+j] = visited[x][y]+1
                    que.append((x+i, y+j))

print(bfs())

 

dp와 최단 거리를 찾기 위해 bfs로 구현

visited에 거리를 넣는다.

4방향으로 탐색한다. 범위내에서 탐색하도록 check 함수를 사용했다.

 

 

👩‍🦰 다른 사람 풀이

 

from sys import stdin
N, M=map(int, stdin.readline().split())
# matrix 배열
matrix=[stdin.readline().rstrip() for _ in range(N)]    # 미로 생성
# 방문경로 배열
visited=[[0]*M for _ in range(N)]
# 좌/우/위/아래 방향 이동
dx, dy=[-1, 1, 0, 0], [0, 0, -1, 1]

# BFS 경로 탐색
# queue 방식 사용
queue=[(0, 0)]
visited[0][0]=1

while queue :
    x, y=queue.pop(0)

    if x == N-1 and y == M-1 :
        # 최종 경로 도착
        print(visited[x][y])
        break

    for i in range(4) :
        nx=x+dx[i]
        ny=y+dy[i]
        if 0 <= nx < N and 0 <= ny < M :
            if visited[nx][ny] == 0 and matrix[nx][ny] == '1' :
                visited[nx][ny]=visited[x][y]+1
                queue.append((nx, ny))

 

bfs로 구현했지만 리스트 사용

 

 

문제 출처 💁‍♀️ 백준

반응형

'coding test' 카테고리의 다른 글

[파이썬] 2606. 바이러스  (0) 2021.05.23
[파이썬] 1743. 음식물 피하기  (0) 2021.05.23
[파이썬] 1303. 전쟁 - 전투  (0) 2021.05.22
[파이썬] 1260. DFS와 BFS  (0) 2021.05.21
[파이썬] 1949. 우수 마을  (0) 2021.05.20