문제
https://school.programmers.co.kr/learn/courses/30/lessons/172928
지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.
- ["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.
- 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
- 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.
공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
- 3 ≤ park의 길이 ≤ 50
- 3 ≤ park[i]의 길이 ≤ 50
- park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
- S : 시작 지점
- O : 이동 가능한 통로
- X : 장애물
- park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
- park는 직사각형 모양입니다.
- 3 ≤ park[i]의 길이 ≤ 50
- 1 ≤ routes의 길이 ≤ 50
- routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
- 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
- routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
- op는 다음 네 가지중 하나로 이루어져 있습니다.
- N : 북쪽으로 주어진 칸만큼 이동합니다.
- S : 남쪽으로 주어진 칸만큼 이동합니다.
- W : 서쪽으로 주어진 칸만큼 이동합니다.
- E : 동쪽으로 주어진 칸만큼 이동합니다.
- 1 ≤ n ≤ 9
- op는 다음 네 가지중 하나로 이루어져 있습니다.
입출력 예
입출력 예 #1
입력된 명령대로 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동하면 [0,0] -> [0,2] -> [2,2] -> [2,1]이 됩니다.
입출력 예 #2
입력된 명령대로라면 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동해야하지만 남쪽으로 2칸 이동할 때 장애물이 있는 칸을 지나기 때문에 해당 명령을 제외한 명령들만 따릅니다. 결과적으로는 [0,0] -> [0,2] -> [0,1]이 됩니다.
입출력 예 #3
처음 입력된 명령은 공원을 나가게 되고 두 번째로 입력된 명령 또한 장애물을 지나가게 되므로 두 입력은 제외한 세 번째 명령만 따르므로 결과는 다음과 같습니다. [0,1] -> [0,0]
나의 풀이
def solution(park, routes):
m, n = len(park), len(park[0])
cur = [0, 0]
def check_is_possible(x, y, dx, dy):
is_dx_movable = True
if y != dy:
is_dx_movable = False
if is_dx_movable:
for i in range(x, dx + 1):
if park[i][y] == 'X':
return False
else:
for i in range(y, dy + 1):
if park[x][i] == 'X':
return False
return True
for i in range(n):
if 'S' in park[i]:
cur = [i, park[i].index('S')]
break
for s in routes:
op, d = s.split()
d = int(d)
x, y = cur[0], cur[1]
if op == 'N':
nx, ny = x - d, y
elif op == 'S':
nx, ny = x + d, y
elif op == 'W':
nx, ny = x, y - d
elif op == 'E':
nx, ny = x, y + d
if 0 <= nx < m and 0 <= ny < n:
if check_is_possible(min(x, nx), min(y, ny), max(x, nx), max(y, ny)):
cur = [nx, ny]
return cur이 문제에서 핵심은 이동 과정 중에 장애물이 하나라도 있는지를 판별하는 것이었다.
먼저, 이동 명령을 읽을 때마다 다음 위치를 계산하여 `nx, ny`에 저장한다.
그다음에는 공원 범위를 벗어나는지 1차로 검사한 뒤,
범위 안이라면 `check_is_possible()`을 이동 경로에 장애물이 있는지를 확인했다.
경로 이동 로직을 통일하기 위해 `check_is_possible()`에 전달하는 네 좌표는
`min()`과 `max()`를 사용해 “작은 좌표 → 큰 좌표” 순으로 전달하도록 했다.
`check_is_possible()` 내부에서는
이동 방향이 x축인지 y축인지를 먼저 판단하고,
해당 방향으로 좌표를 하나씩 움직이며 'X'가 있는지를 검사한다.
중간에 장애물이 발견되면 즉시 `False`를 반환하고,
끝까지 장애물이 발견되지 않는다면 `True`를 반환한다.
이 값이 `True`라면 경로에 장애물이 없는 것이므로
현재 위치 `cur`을 새 좌표로 갱신해 이동을 완료한다.
다른 사람의 풀이
def solution(park, routes):
dirs = {'N': (-1, 0), 'S': (1, 0), 'W': (0, -1), 'E': (0, 1)}
m, n = len(park), len(park[0])
answer = [0, 0]
for i in range(n):
if 'S' in park[i]:
answer = [i, park[i].index('S')]
break
for route in routes:
dir, dist = route.split()
dist = int(dist)
x, y = answer[0], answer[1]
for i in range(1, dist + 1):
nx, ny = x + (dirs[dir][0] * i), y + (dirs[dir][1] * i)
if nx < 0 or nx >= m or ny < 0 or ny >= n:
break
if park[nx][ny] == 'X':
break
else:
answer = [nx, ny]
return answer각 방향에 대해 이동해야 하는 변화량 `(dx, dy)`를 딕셔너리에 미리 저장해서
`dirs[dir]`에 `i`를 곱해가며 좌표를 계산하기 때문에 방향별 이동 계산이 훨씬 단순해졌다.
이에 따라 이동해야 할 거리만큼 1칸씩 직접 이동해 보며 경로 전체를 바로바로 검사할 수 있었다.
또한 for-else 구문을 활용해 경로 전체를 안전하게 통과했을 때만 else문이 실행되어 새로운 좌표로 이동한다.