[Algorithm] Programmers : 방문 길이 by Python

엄희관·2020년 12월 29일
0

Algorithm

목록 보기
44/128
post-thumbnail

[문제 바로가기] https://programmers.co.kr/learn/courses/30/lessons/49994

📌문제 설명

게임 캐릭터를 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 이하의 자연수입니다.

입출력 예

입출력 예 설명
입출력 예 #1
문제의 예시와 같습니다.

입출력 예 #2
문제의 예시와 같습니다.


💡 문제 풀이

좌표평면상에서 상하좌우로 움직일 때 중복방문 하지 않은 총 길이를 구하면 되는 문제다.

문제에서도 좌표평면의 좌표로서 이동경로를 나타냈기 때문에 인덱스(행, 열)를 이용하여 문제 내용을 구현하였다.
※ 좌표를 통해서 중복방문 여부만 파악하면 되기 때문에 좌표평면에 해당하는 이중리스트는 만들 필요가 없다.

step 1)
좌표평면에서 초기 위치는 (0, 0)이지만 11 x 11의 좌표평면(이중리스트)을 구현했을 때 중심위치는 (5, 5)이다.

따라서, 행과 열의 위치를 (5, 5)로 초기 설정하였다.

  • r : 행의 위치
  • c : 열의 위치
  • used : 방문한 좌표를 담는 리스트

step 2)
dirs에 있는 문자들에 따라 좌표를 움직인다.
이 때, 좌표가 좌표평면 상에서 벗어나면 안되기 때문에 조건문을 이용하여 벗어나는 경우 continue를 해준다.

step 3)
좌표 평면상에서 벗어나지 않으면 방문여부를 파악한다.

  • check : 방문 여부를 파악하는 함수
  • nr : 이동후 행좌표
  • nc : 이동후 열좌표

방문여부는 간단하게 "이동전 좌표 → 이동후 좌표" 또는 "이동후 좌표 → 이동전 좌표"에 해당하는 이동이 없었으면 방문하지 않았다는 뜻이므로

[[r, c], [nr, nc]] 또는 [[nr, nc], [r, c]]가 used에 없으면 [[r, c], [nr, nc]]를 used에 추가하였다.

방문여부를 파악한 후 UDRL에 맞는 좌표를 이동한다.

dirs의 문자열 반복을 모두 마치면 used에는 처음 걸어본 길의 좌표들이 담겨있기 때문에 used 리스트의 길이를 return해주면 된다.


코드는 다음과 같다.

def check(r, c, nr, nc, used):
    if [[r, c], [nr, nc]] not in used and [[nr, nc], [r, c]] not in used:
        used.append([[r, c], [nr, nc]])
    return used

def solution(dirs):
    r, c = 5, 5
    used = []
    for dir in dirs:
        if dir == "U":
            if r-1 < 0:continue
            used = check(r, c, r-1, c, used)
            r -= 1
        elif dir == "D":
            if r+1 > 10:continue
            used = check(r, c, r+1, c, used)
            r += 1
        elif dir == "R":
            if c+1 > 10:continue
            used = check(r, c, r, c+1, used)
            c += 1
        else:
            if c-1 < 0:continue
            used = check(r, c, r, c-1, used)
            c -= 1
    return len(used)
profile
허브

0개의 댓글