Lv.1 공원 산책

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

프로그래머스

 

머리가 안돌아가서 정답보고 풀었습니다.

⭐️⭐️추후에 정답보지말고 혼자 다시 풀어볼것

def solution(park, routes):

    # 초기 위치
    a = 0
    b = 0

    # 시작 위치 찾기
    for i in range(len(park)):
        for j in range(len(park[i])):
            if park[i][j] == 'S':
                a = j
                b = i
                break

    for route in routes:
        x = a
        y = b
        # route의 움직이는 거리안에서
        for step in range(int(route[2])):
            # 동쪽
            if route[0] == 'E' and x != len(park[0])-1 and park[y][x+1] != 'X':
                x += 1
                if step == int(route[2])-1:
                    a = x
            # 서쪽
            elif route[0] == 'W' and x != 0 and park[y][x-1] != 'X':
                x -= 1
                if step == int(route[2])-1:
                    a = x
            # 남쪽
            elif route[0] == 'S' and y != len(park)-1 and park[y+1][x] != 'X':
                y += 1
                if step == int(route[2])-1:
                    b = y
            # 북쪽
            elif route[0] == 'N' and y != 0 and park[y-1][x] != 'X':
                y -= 1
                if step == int(route[2])-1:
                    b = y

    return [b, a]