[레벨1][파이썬] 공원 산책

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

프로그래머스

 

장애물을 만나면 이동한 위치를 롤백해야 한다. 이러한 로직을 처리함에 있어 func 함수를 정의하여 문제를 해결했다. 최초에는 해당 로직을 함수로 빼지 않고 작성하여 제출했는데 "코드 실행"에서의 테스트 케이스는 통과하였으나 제출 시에는 통과하지 못한 테스트 케이스가 다수 발생했었다. 이동 로직을 함수로 정의하여 재사용하는 방식으로 적용하니 간결한 코드와 통과가 가능했다.

def solution(park, routes):
    start = []
    x = []
    w_s = len(park[0])
    h_s = len(park)
    
    for h, s in enumerate(park):
        for w, v in enumerate(s):
            if v == "S":
                start = [h, w]
            if v == "X":
                x.append([h, w])
    
    def func(lst):
        temp = start[:]
        lst[1] = int(lst[1])
        for _ in range(lst[1]):
            if lst[0] == 'W':
                temp[1] -= 1
            if lst[0] == 'E':
                temp[1] += 1
            if lst[0] == 'N':
                temp[0] -= 1
            if lst[0] == 'S':
                temp[0] += 1
            if temp[0] < 0 or temp[0] >= h_s:
                return
            if temp[1] < 0 or temp[1] >= w_s:
                return
            if temp in x:
                return
        return temp

    for i in routes:
        r = i.split()
        coord = func(r)
        if coord:
            start = coord

    return start