Algorithm/이것이 코딩 테스트다/구현/상하좌우

yellow·2021년 4월 20일

알고리즘 문제

목록 보기

6/58

📖문제

여행가 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 크기의 정사각형 공간을 벗어나는 움직임은 무시된다.
계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오

입력 조건

첫째 줄에 공간의 크기를 나타내는 N이 주어진다. (1 <= N <= 100)

둘째 줄에 여행가 A가 이동할 계획서 내용이 주어진다.
(1 <= 이동 횟수 <= 100)

출력 조건

첫째 줄에 여행가 A가 최종적으로 도착할 지점의 좌표(X, Y)를 공백으로 구분하여 출력한다.

📝풀이과정

계획서를 처음부터 읽어들이면서 어떤 문자인지 검사
문자가 L이라면 현재 열이 1이라면 무시, 아니면 열-1
문자가 R이라면 현재 열이 N이라면 무시, 아니면 열+1
문자가 U라면 현재 행이 1이라면 무시, 아니면 행-1
문자가 D라면 현재 행이 N이라면 무시, 아니면 행+1

⌨코드1 (내 코드)

# 입력 조건
n = int(input())
plan = list(input().split())

# 처음 시작 위치는 (1, 1)
row, column = 1, 1
for d in plan:
  # 문자가 L이라면 현재 열이 1이라면 무시, 아니면 열-1
  if d == 'L':
    if column != 1:
      column -= 1
  # 문자가 R이라면 현재 열이 N이라면 무시, 아니면 열+1    
  elif d == 'R':
    if column != n:
      column += 1
  # 문자가 U라면 현재 행이 1이라면 무시, 아니면 행-1
  elif d == 'U':
    if row != 1:
      row -= 1
  # 문자가 D라면 현재 행이 N이라면 무시, 아니면 행+1
  elif d == 'D':
    if row != n:
      row += 1

# 최종 도착지 출력
print(row, column)

⌨코드2 (책 예시 코드)

이 문제처럼 x, y좌표를 움직여야 하는 코드라면 튜플에 미리 담아놓고 인덱스를 이용하는 것을 생각하자!!

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)