문제 설명
지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.
["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.
공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
주요 특징
S : 시작 지점
O : 이동 가능한 통로
X : 장애물
park는 직사각형 모양입니다.
로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
op는 다음 네 가지중 하나로 이루어져 있습니다.
N : 북쪽으로 주어진 칸만큼 이동합니다.
S : 남쪽으로 주어진 칸만큼 이동합니다.
W : 서쪽으로 주어진 칸만큼 이동합니다.
E : 동쪽으로 주어진 칸만큼 이동합니다.
1 ≤ n ≤ 9
입출력 예
park routes result
["SOO","OOO","OOO"]["E 2","S 2","W 1"] [2,1]["SOO","OXX","OOO"] ["E 2","S 2","W 1"][0,1]
["OSO","OOO","OXO","OOO"]["E 2","S 3","W 1"] [0,0]
주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.
문제 풀이
park(2차원)배열: 공원의 지형(O,X로 이루어짐)
S:시작위치, O: 갈 수 있는 길, X: 장애물
routes배열: 강아지가 움직일 루트, 한번 움직일때 지형에 문제있으면 다음 동작으로 넘어감
E:동쪽, S:남쪽, W:서쪽
result배열: 모든 동작을 수행한 후의 위치
방법
1. routes배열을 돌면서 동작을 수행하는데
해당 방향 & 칸수 사이에 공원 벗어나거나 장애물 => 다음 동작 수행
2. 그렇지 않으면 해당 동작을 수행
3. 마지막엔 현재 위치 result배열에 담아서 반환
def solution(park, routes):
#정사각형 배열이 아니라 직사각형 배열이 될 수도 있음 <- 놓치면 안됨
length_x = len(park)#행의 크기
length_y = len(park[0])#열의 크기
answer = []
x, y = (0, 0)
#시작 위치 설정
for index_x, arr in enumerate(park):
for index_y, char in enumerate(arr):
if(isinstance(char, str) and char == 'S'):
#x = index_x, y = index_y <- 이거 안됨!
x, y = (index_x, index_y)
def check(dir, n):
if dir == 'E': #동쪽, 열을 움직여야 함
for index in range(y, y+n+1):
if park[x][index] == 'X':
return False
return True
elif dir == 'S': #남쪽, 행을 움직여야 함
for index in range(x, x+n+1):
if park[index][y] == 'X':
return False
return True
elif dir == 'W': #서쪽, 열을 움직여야 함
for index in range(y, y-n-1, -1):
if park[x][index] == 'X':
return False
return True
elif dir == 'N': #북쪽, 행을 움직여야 함
for index in range(x, x-n-1, -1):
if park[index][y] == 'X':
return False
return True
for route in routes:#2차원 배열 routes를 돌면서
arr = route.split(' ')
dir = arr[0]#방향
num = int(arr[1])#갈 칸수
#공원 벗어나거나 중간에 장애물 있을때
#동쪽일때
if(dir == 'E'):
if(y + num > length_y - 1 or not check(dir, num)):
continue
else:
y = y + num
#남쪽일때
elif(dir == 'S'):
if(x + num > length_x - 1 or not check(dir, num)):
continue
else:
x = x + num
#서쪽일때
elif(dir == 'W'):
if(y - num < 0 or not check(dir,num)):
continue
else:
y = y - num
#북쪽일때
elif(dir == 'N'):
if(x - num < 0 or not check(dir, num)):
continue
else:
x = x - num
answer.append(x)
answer.append(y)
return answer후기)
문제를 꼼꼼히 읽자. 정사각형 케이스만 있는 줄 알았는데 직사각형 케이스도 있었다.
