Algorithm/문제풀이

구현 - 상하좌우

Python Developer 2025. 3. 4. 01:40

상하좌우 (Python)

📝 문제 설명

여행가 A 는 N * X 크기의 정사각형 공간 위에 서 있습니다.
이 공간은 1 x 1 크기의 정사각형으로 나누어져 있습니다.
가장 왼쪽 위 좌표는 (1,1)이며 가장 오른쪽 아래 좌표는 (N, N) 에 해당합니다.
여행가 A 는 상, 하, 좌, 우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1,1)
입니다. 우리 앞에는 여행가 A 가 이동 할 계획이 적힌 계획서가 놓여 있습니다.
계획서에는 하나의 줄에 띄어쓰기를 기준으로 하여 L,R,U,D 중 하나의 문자가
반복적으로 적혀 있습니다. 각 문자의 의미는 다음과 같습니다.
L: 왼쪽으로 한 칸 이동
R: 오른쪽으로 한 칸 이동
U: 위로 한 칸 이동
D: 아래로 한 칸 이동
이때 여행가 A 가 N x N 크기의 정사각형 공간을 벗어나는 움직임은 무시됩니다.


🔽 입력 예시

N = 5
plans = ['R', 'R', 'R', 'U', 'D', 'D']

🔼 출력 예시

1 4


💡 문제 해결 접근

move_types = ['L', 'R', 'U', 'D'], move_types 의 원소에 해당하는
방향 좌표 dx,dy 설정 이후 구현
dx = [0, 0, 1, -1], x 좌표 이동
dy = [-1, 1, 0, 0], y 좌표 이동

def solution(N, plans):
    x, y = 1, 1
    dx = [0, 0, 1, -1]
    dy = [-1, 1, 0, 0]
    move_types = ['L', 'R', 'U', 'D']
    for plan in plans:
        idx = move_types.index(plan)
        nx = x + dx[idx]
        ny = y + dy[idx]
        if nx < 1 or ny < 1 or nx > N or ny > N:
            continue
        x, y = nx, ny
    return x, y

🏆 정답 코드

def 모범답안(N, plans):
    x, y = 1, 1

    dx = [0, 0, 1, -1]
    dy = [-1, 1, 0, 0]

    move_types = ['L', 'R', 'U', 'D']
    for plan in plans:
        for i in range(len(move_types)):
            if plan == move_types[i]:
                nx = x + dx[i]
                ny = y + dy[i]
        if nx < 1 or ny < 1 or nx > N or ny > N:
            continue
        x, y = nx, ny

🔥 느낀 점

  • 유형만 알면 단순한 문제
  • 모범 답안에서는 이중for문과 if 문으로 구현하는데 이중 for 문 보다 index 찾아서 구현 하는게 효율이 더 좋은게 아닌가 싶지만 유의미한 차이도 아닌듯 싶다. O(4N) vs O(N) 차이 인가