[Python][문제풀이] 상하좌우

0

이취코

목록 보기
4/8

난이도 ●○○ | 시간 제한 1초 | 메모리 제한 128MB

🚀 문제

여행가 A는 N x N 크기의 정사각형 공간 위에 서 있다.
이 공간은 1 x 1 크기의 정사각형으로 나누어져 있다.
가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다.
여행가 A는 상,하,좌,우 방향으로 이동할 수 있으며 시작 좌표는 항상 (1,1)이다.

우리 앞에는 여행가 A가 이동할 계획이 적힌 계획서가 놓여 있다.
계획서에는 하나의 줄에 띄어쓰기를 기준으로하여 L,R,U,D 중 하나의 문자가 반복적으로 적혀 있다.
각 문자의 의미는 다음과 같다.

  • L: 왼쪽으로 한칸 이동
  • R: 오른쪽으로 한칸 이동
  • U: 위로 한칸 이동
  • D: 아래로 한칸 이동

이때 여행가 A가 N X N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.

ex

입력

5
R R R U D D

출력

3 4

📝 내 풀이

n = int(input())
data = list(map(str, input().split()))
x, y = 1, 1
for i in data:
    if i == 'L' and y > 1:
        y -= 1
    elif i == 'R' and y < n+1:
        y += 1
    elif i == 'U' and x > 1:
        x -= 1
    elif i == 'D' and x < n+1:
        x += 1
print(x, y)

해설

이 문제를 요구사항대로 구현하면 연산 횟수는 이동 횟수에 비례하게 된다. 예를 들어 이동 횟수가 N번인 경우 시간 복잡도는 O(N)이다. 따라서 이 문제의 시간 복잡도는 매우 넉넉한 편이다.
이러한 문제는 일련의 명령에 따라서 개체를 차례대로 이동시킨다는 점에서 시뮬레이션 유형으로 분류되며 구현이 중요한 대표적인 문제 유형이다. 다만 , 알고리즘 교재나 문제 풀이 사이트에 따라서 다르게 일컬을 수 있으니 코딩 테스트에서의 시뮬레이션 유형, 구현 유형, 완전 탐색 유형은 서로 유사한 점이 많다는 정도로만 기억하자.
코딩 테스트나 알고리즘 대회에서 가장 난이도가 낮은 1 ~ 2 번 문제는 대부분 그리디 알고리즘이나 구현 문제이다. 이 두 유형이 논리적 사고력을 확인할 수 있는 가장 기본 난이도의 문제로 적합하기 때문이다. 난이도가 낮은 만큼 합격을 좌우하는 중요한 문제이기도 하다.

# N을 입력받기
n = int(input())
x, y = 1, 1
plans = input().split()
# L, R, U, D에 따른 이동 방ㅇ향
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
print(x, y)

문제 출처 : 이것이 취업을 위한 코딩테스트다 (저자 : 나동빈)

0개의 댓글