[프로그래머스] 공원 산책

문제 설명

지나다니는 길을 '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는 직사각형 모양입니다.
1 ≤ routes의 길이 ≤ 50
routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
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]

입출력 예 설명

입출력 예 #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]

 

 

풀이

 

코드가 길어지고 조건문과 반복문이 많아 시간복잡도에 통과될 지 고민이 됐지만 다행히도 쉽게 통과됐다.!

 

어떻게 하면?
이번 문제는 독해 문제로 문제를 잘 이해하여 필요한 것이 무엇인지 파악하기만 하면! 

우선 필요한 것은 행/열의 길이, 시작 포인트 S의 행/열 위치이다. 단순 반복문으로 해당 값들을 찾은 후

E : col+1 , W : col -1, S: row+1,  N:row-1 라는 조건문을 둬 해당 값들을 이동 , 이동시  

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
• 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

이 두 조건문을 체크하여  row,col을 이동시키면 된다!

def check(park,row,col): # 체크함수 
    # 행/열이 벗어나지 않았는지 , X에 마주하지 않았는지 체크
    if(row<0 or col<0 or row>=len(park) or col>=len(park[0]) or park[row][col]=="X"):
        return False
    else:
        return True
        
def move(park,row,col,location,count):
    # row,col을 저장한 임시 변수 저장
    temp_row=row
    temp_col=col
    for i in range(count):
        if location=="E":
            col+=1
        elif location=="W":
            col-=1
        elif location=="S":
            row+=1
        elif location=="N":
            row-=1
        
        # 명령을 무시하기 위해 임시 저장변수를 반환
        if(check(park,row,col)==False):
            return temp_row,temp_col
    return row,col


def solution(park, routes):

    row=len(park)
    col=len(park[0])
    
    # 시작 위치 "S" 찾기
    for index_row,p in enumerate(park):
        for index_col,pp in enumerate(p):
            if(pp in "S"):
                start_row=index_row
                start_col=index_col
                break
    
    for route in routes:
        s=route.split(" ")
        location=s[0]
        count=int(s[1])
        start_row,start_col=move(park,start_row,start_col,location,count)

    answer = []
    answer.append(start_row)
    answer.append(start_col)

    return answer