상하좌우 (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) 차이 인가
'Algorithm > 문제풀이' 카테고리의 다른 글
BFS & DFS - 음료수 얼려 먹기 (0) | 2025.03.05 |
---|---|
구현 - 시각 (0) | 2025.03.04 |
그리디 알고리즘 - 모험가 길드 (0) | 2025.03.03 |
그리디 알고리즘 - 곱하기 혹은 더하기 (0) | 2025.03.03 |
그리디 알고리즘 - 1이 될때까지 (0) | 2025.03.03 |