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

https://school.programmers.co.kr/learn/courses/30/lessons/172928

프로그래머스

 

나의 코드(성공)

처음에는 runtime 에러 때문에 많이 헤맸다. 하나하나 찾아보니 시작 위치 탐색에서 h, w를 범위로 설정하였기 때문...

def solution(park, routes):
    answer = []
    h, w = len(park)-1, len(park[0])-1
    
    # 시작 위치 탐색
    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == "S":
                answer.append(i)
                answer.append(j)
                break
    
    for i in routes:
        direction, num = i[0], int(i[2])

        # 이동 좌표
        x, y = answer

        # 한 칸씩 이동, 현재 위치가 끝 부분이면 무시, 장애물이면 무시
        for j in range(num):
            if direction == "E" and y != w and park[x][y+1] != "X":
                y += 1
                if j == num -1:
                    answer[1] = y
            elif direction == "W" and y != 0 and park[x][y-1] != "X":
                y -= 1
                if j == num -1:
                    answer[1] = y
            elif direction == "S" and x != h and park[x+1][y] != "X":
                x += 1
                if j == num -1:
                    answer[0] = x
            elif direction == "N" and x != 0 and park[x-1][y] != "X":
                x -= 1
                if j == num -1:
                    answer[0] = x
     return answer