문제
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 |