지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.
["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.
주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다. 공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.
공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
내 풀이
def solution(park, routes):
answer = []
park_array = []
obstacle = []
w, h = len(park[0]), len(park)
x , y = 0, 0
op={'N':(-1,0), 'S':(1,0), 'W':(0,-1),'E':(0,1)}
#시작위치
for i in range(h):
for j in range(w):
if park[i][j] == 'S':
y , x = i, j
#이동
for route in routes:
di, dist = route.split()
dist = int(dist)
now_x, now_y = x, y
for i in range(dist):
nx = x + op[di][1]
ny = y + op[di][0]
if 0<=nx<=w-1 and 0<=ny<=h-1 and park[ny][nx] != 'X':
x, y = nx, ny
else:
x, y = now_x, now_y
break
print()
answer = (y,x)
return answer
op를 선언하여 각 방향으로 한칸씩 움직여야 장애물이 그 자리에 있는지 없는지 확인 가능하다.
