[프로그래머스] 방문 길이 - Python3

프로그래머스에 level2 문제가 새로 나왔길래 풀어보았습니다. :)

 

1. 문제 설명

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.

• U: 위쪽으로 한 칸 가기

• D: 아래쪽으로 한 칸 가기

• R: 오른쪽으로 한 칸 가기

• L: 왼쪽으로 한 칸 가기

캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

예를 들어, "ULURRDLLU"로 명령했다면

• 1번 명령어부터 7번 명령어까지 다음과 같이 움직입니다.

• 8번 명령어부터 9번 명령어까지 다음과 같이 움직입니다.

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)

단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.

예를 들어, "LULLLLLLU"로 명령했다면

• 1번 명령어부터 6번 명령어대로 움직인 후, 7, 8번 명령어는 무시합니다. 다시 9번 명령어대로 움직입니다.

이때 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다.

명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여 return 하는 solution 함수를 완성해 주세요.

제한사항

• dirs는 string형으로 주어지며, 'U', 'D', 'R', 'L' 이외에 문자는 주어지지 않습니다.
• dirs의 길이는 500 이하의 자연수입니다.

입출력 예

dirs                                                                         answer

"ULURRDLLU"	7
"LULLLLLLU"	7

 

 

2. 접근 방법 및 풀이

- 처음 (0,0)의 좌표에 위치한 캐릭터를 -> 11x11 2차원 배열의 (5,5)에 위치하고 있다고 생각하고 구현했습니다.

 

- 캐릭터가 움직이는 길이 row에 해당하는 가로줄과 col에 해당하는 세로줄 이기 때문에 가로줄과 세로줄에 해당하는 리스트를 각각 만들어서 그 줄에 해당하는 인덱스를 방문했는지를 살펴보고, 처음 방문했다면 answer를 증가시켜주는 방식으로 구현했습니다.

 

- 범위를 알맞게 예외처리 해주지 않으면 무수히 발생하는 런타임 에러를 볼 수 있습니다....

 

- 가장 헤맸던 부분이 처음 캐릭터가 있는 지점이 (5,5) 라고 했을 때, 가로줄의 경우 캐릭터 점의 우측에 있는 가로줄, 세로줄의 경우 캐릭터 점의 아래쪽 세로줄이 처음 스타트 지점이기 때문에 'D'나 'R'이 나오게 되면 원하지 않는 결과가 나오는 부분이었는데, 인덱스가 증가하는 부분에서는 그 인덱스를 먼저 방문처리 한 후 증가시켜주는 식으로 하면 해결됩니다! (1시간 헤맴...)

 

def solution(dirs):
    answer = 0
    
    row, col = 5, 5  # 처음 캐릭터가 있는 위치
 
    row_visited = [[False] * 11 for _ in range(11)] # 가로줄
    col_visited = [[False] * 11 for _ in range(11)] # 세로줄
 
    for command in dirs:
        if command == 'U':  # 위로 한 칸 이동
            if row == 0:    # 맨 위면
                continue
 
            row -= 1        # 한 칸 위로 이동하고 
            if col_visited[row][col]: # 이동했던 세로줄을 이미 방문했다면
                continue
            else:
                col_visited[row][col] = True # 아니면 방문처리
                answer += 1 # answer 증가
 
        elif command == 'L':
            if col == 0:
                continue
 
            col -= 1
            if row_visited[row][col]:
                continue
            else:
                row_visited[row][col] = True
                answer += 1
 
        # 'R'과 'D'는 증가 연산이기 때문에 방문처리 한 후 값을 증가시켜야 함        
        elif command == 'R':
            if col == 10:
                continue
 
            if row_visited[row][col]:
                col += 1
                continue
            else:
                row_visited[row][col] = True 
                col += 1
                answer += 1
 
        else:
            if row == 10:
                continue
            
            if col_visited[row][col]:
                row += 1
                continue
            else:
                col_visited[row][col] = True
                row += 1
                answer += 1
 
    return answer