✔ 지문

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

• ["방향 거리", "방향 거리" … ]

예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

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

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

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

✔ 최초풀이

def solution(park, routes):

    a,b = 0,0
    l_b = len(park[0])-1
    l_a = len(park)-1
    
    for i,row in enumerate(park):
        if 'S' in row:
            a = i
            b = row.find('S')
            break;
    for route in routes :
        isX =  False
        r = int(route[2:])
        if route[0] == 'E': #  동   
            if b+int(r) > l_b : #범위 초과 시
                continue;
            for i in range(1, r+1): #X체크
                if park[a][b+i] == 'X' :
                    isX = True
                    break;
            if not isX :
                b += r
                
        elif route[0] == 'W': # 서
            if b - int(r) < 0 :
                continue;
            for i in range(1,r+1):
                if park[a][b-i] == 'X':
                    isX = True
                    break;
            if not isX :
                b -= r
        elif route[0] == 'S': # 남
            if a + int(r) > l_a:
                continue;
            for i in range(1,r+1):
                if park[a+i][b] == 'X' :
                    isX =True
                    break;
            if not isX :
                a += r
        elif route[0] == 'N': # 북
            if a - int(r) < 0:
                continue;
            for i in range(1,r+1) :
                if park[a-i][b] == 'X':
                    isX =True
                    break;
                    
            if not isX :
                a -= r
            
    return [a,b]

처음엔 직관적으로 일단 문제를 풀었다. 

근데 중복되는 코드가 너무 많은거 같아서 리팩토링을 하기로 했다. 

def solution(park, routes):

    a,b = 0,0
    l_b = len(park[0])-1
    l_a = len(park)-1
    vector = {'E': (0,1), "W" : (0,-1), 'S':(1,0), "N":(-1,0)}
    
    # 초기화 
    for i,row in enumerate(park):
        if 'S' in row:
            a = i
            b = row.find('S')
            break;
            
    for route in routes :
        r = int(route[2:])
        da, db = vector[route[0]]
        na, nb = a+da*r, b+db*r
        if not(0 <= na <= l_a and 0 <= nb <= l_b): #범위 초과시
            continue;
        
        isX =  False
        for i in range(1,r+1):
            ta, tb = a+da*i, b+db*i
            if park[ta][tb] == 'X':
                isX = True
                break;
                
        if not isX :
            a, b = na, nb       
            
    return [a,b]

벡터를 딕셔너리로 묶어서 중복로직을 정리했다.