|| 문제 설명 ||
1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.
미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.
제한사항
- 5 ≤ maps의 길이 ≤ 100
- 5 ≤ maps[i]의 길이 ≤ 100
- maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
- S : 시작 지점
- E : 출구
- L : 레버
- O : 통로
- X : 벽
- 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
- 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.
입출력 예mapsresult
| ["SOOOL","XXXXO","OOOOO","OXXXX","OOOOE"] | 16 |
| ["LOOXS","OOOOX","OOOOO","OOOOO","EOOOO"] | -1 |
입출력 예 설명
입출력 예 #1
주어진 문자열은 다음과 같은 미로이며
다음과 같이 이동하면 가장 빠른 시간에 탈출할 수 있습니다.
4번 이동하여 레버를 당기고 출구까지 이동하면 총 16초의 시간이 걸립니다. 따라서 16을 반환합니다.
입출력 예 #2
주어진 문자열은 다음과 같은 미로입니다.
시작 지점에서 이동할 수 있는 공간이 없어서 탈출할 수 없습니다. 따라서 -1을 반환합니다.
틀린풀이 1
계속 이상하게 문제를 풀어서 이 문제만 3시간은 잡고 있었던거 같다ㅠㅠ
일단 처음에 했던 방식은 아래와 같다
from collections import deque
def solution(maps):
answer = 0
n = len(maps)
m = len(maps[0])
dh = [1, -1, 0, 0]
dw = [0, 0, 1, -1]
vis = []
visForE = []
dis = []
s = []
e = []
# visit 만들기
for i in range(n):
arr = []
for j in range(m):
arr.append(0)
vis.append(arr)
visForE.append(arr)
dis.append(arr)
# 시작점 찾기
for i in range(n):
for j in range(m):
if maps[i][j] == "S":
s.append(i)
s.append(j)
# 탈출점 찾기
for i in range(n):
for j in range(m):
if maps[i][j] == "E":
e.append(i)
e.append(j)
# 레버까지의 최단거리 + 레버 -> 탈출구 까지의 최단거리
time = 0
queue = deque()
queue.append(s)
equeue = deque()
while queue:
h, w = queue.popleft()
vis[h][w] = 1 # 방문 처리
if maps[h][w] == "L":
equeue.append((h, w, 0))
break
for i in range(4):
nh = h + dh[i]
nw = w + dw[i]
if nh < 0 or nh >= n or nw < 0 or nw >= m:
continue
if maps[nh][nw] == "X":
continue
if vis[nh][nw] == 1:
continue
if vis[nh][nw] == 0:
queue.append((nh, nw))
time += 1
print(time)
print(queue)
print(equeue)이렇게 레버까지의 걸린 시간을 구한 후에 레버에서 탈출구까지 걸리는 시간을 구할려고 했다. 하지만 이 방식은 탐색을 진행할 때마다 전역 변수인 time에 +1을 해주기 때문에 탐색하는 모든 시간을 구하므로 옳지 않다.
그래서 큐에 비용까지 포함해서 다시 코드를 작성했다.
틀린 풀이 2
from collections import deque
def solution(maps):
answer = 0
n = len(maps)
m = len(maps[0])
dh = [1, -1, 0, 0]
dw = [0, 0, 1, -1]
lastcost = 0
vis = []
visForE = []
s = []
# visit 만들기
for i in range(n):
arr = []
for j in range(m):
arr.append(0)
vis.append(arr)
visForE.append(arr)
# 시작점 찾기
for i in range(n):
for j in range(m):
if maps[i][j] == "S":
s.append(i)
s.append(j)
# 레버까지의 최단시간 구하기
h, w = s
vis[h][w] = 1
queue = deque()
queue.append((h, w, 0))
while queue:
h, w, cost = queue.popleft()
if maps[h][w] == "L":
queue.append((h, w, cost))
break
for i in range(4):
nh = h + dh[i]
nw = w + dw[i]
if nh < 0 or nh >= n or nw < 0 or nw >= m:
continue
if maps[nh][nw] == "X":
continue
if vis[nh][nw] == 1:
continue
if vis[nh][nw] == 0:
queue.append((nh, nw, cost + 1))
vis[nh][nw] = 1
# 레버에서 탈출점까지의 최단 시간 구하기
while queue:
x, y, cost = queue.popleft()
if maps[x][y] == "E":
return cost
break
for i in range(4):
nh = x + dh[i]
nw = y + dw[i]
if nh < 0 or nh >= n or nw < 0 or nw >= m:
continue
if maps[nh][nw] == "X":
continue
if vis[nh][nw] == 1:
continue
if vis[nh][nw] == 0:
queue.append((nh, nw, cost + 1))
vis[nh][nw] = 1
## 비용으로 생각해서 풀자 전역적으로 time을 지정해서
## 풀려고 하니까 최단 거리가 아니라
## 모든 지점을 탐색하는데 걸린 시간이 출력되어서 계속 못 풀었다
return -1이 경우는 무조건 레버를 찾을 수 있다는 가정하에 다음 단계로 넘어가기 때문에 정답을 맞히지 못한건 같다 그래서 bfs 함수를 따로 만들어서 (시작점, 목표점, maps)를 입력해 cost 값을 받는 방식으로 바꿔보기로 했다.
옳은 풀이
from collections import deque
def bfs(start, end, maps):
h, w = start
eh, ew = end
dh = [1, -1, 0, 0]
dw = [0, 0, 1, -1]
n = len(maps)
m = len(maps[0])
visit = []
for i in range(n):
arr = []
for j in range(m):
arr.append(0)
visit.append(arr)
que = deque()
que.append((h, w, 0))
visit[h][w] = 1
while que:
h, w, cost = que.popleft()
if h == eh and w == ew:
return cost
for i in range(4):
nh = h + dh[i]
nw = w + dw[i]
if nh < 0 or nh >= n or nw < 0 or nw >= m:
continue
if maps[nh][nw] == "X":
continue
if visit[nh][nw] == 1:
continue
if visit[nh][nw] == 0:
que.append((nh, nw, cost + 1))
visit[nh][nw] = 1
return -1
def solution(maps):
s = []
e = []
l = []
n = len(maps)
m = len(maps[0])
for i in range(n):
for j in range(m):
if maps[i][j] == "S":
s.append(i)
s.append(j)
for i in range(n):
for j in range(m):
if maps[i][j] == "L":
l.append(i)
l.append(j)
for i in range(n):
for j in range(m):
if maps[i][j] == "E":
e.append(i)
e.append(j)
path1 = bfs(s, l, maps)
path2 = bfs(l, e, maps)
if path1 != -1 and path2 != -1:
return path1 + path2
return -1
함수를 따로 만들어서 시작점과 끝점을 정하고 각각 따로 bfs를 돌려서 각 비용을 계산하고 둘다 찾아진다면 두 비용을 합친 것을 리턴하고 아니라면 -1을 리턴해 마무리 해준다.
항상 차분하게 문제를 보고 신중하게 생각하자
'개인 공부 (전공 공부, 알고리즘 공부 등) > 프로그래머스' 카테고리의 다른 글
| [프로그래머스] Lv.2 다리를 지나는 트럭, 스택/큐 문제를 풀 때 팁 (2) | 2024.04.20 |
|---|---|
| 프로그래머스 알고리즘 고득점 Kit - 해시(Level1 모두 Level2 하나) (2) | 2024.02.29 |
| 코딩테스트를 위한 파이썬 기초 문법 (4) | 2024.02.29 |