프로그래머스: 방문 길이

문제

코딩테스트 연습 - 방문 길이

 

문제 풀이

 캐릭터가 [0, 0] 좌표에서 시작해 명령에 따라 상, 하, 좌, 우로 이동한다. 중복하여 이동한 거리와 범위 밖으로 이동한 경우를 제외하여 이동한 거리의 합을 계산하여 반환하는 문제이다. 문제를 풀기 위해서는 딕셔너리를 통해 `U, D, L, R`로 이동할 경우 변화하는 좌표를 선언한 후, 범위에 따라 방문하지 않은 경우 `visited`에 추가하면 된다.

 

 문제에서 구하고자 하는 답은 `visteid // 2`가 된다. 이와 같은 이유는 이동할 때는 출발한 곳과 도착한 곳 양방향으로 기록하여야 한다. 즉 현재 좌표가 `[x, y]` 다음 좌표가 `[next_x, next_y]`라면 `[x, y, next_x, next_y], [next_x, next_y, x, y]`를 기록하여야 한다.

 

코드

move = {
    'U': (-1, 0),
    'D': (1, 0),
    'L': (0, -1),
    'R': (0, 1)
}


def solution(dirs):
    x, y = 0, 0
    visited = set()

    for command in dirs:
        dx, dy = move[command]
        next_x, next_y = x + dx, y + dy

        if 0 <= abs(next_x) <= 5 and 0 <= abs(next_y) <= 5:
            if (x, y, next_x, next_y) not in visited:
                visited.add((x, y, next_x, next_y))
                visited.add((next_x, next_y, x, y))
            x, y = next_x, next_y

    return len(visited) // 2