[Algorithm] (이코테) 상하좌우- 파이썬

교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 3 구현
실전문제 4-1 상하좌우 110p

상하좌우

문제

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

계획서에는 하나의 줄에 띄어쓰기를 기준으로 L, R, U, D 중 하나의 문자가 반복적으로 적혀있다.
각 문자의 의미는 다음과 같다

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

이때 여행가 A가 N × N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다
예를 들어 (1, 1)의 위치에서 L 혹은 U를 만나면 무시된다
다음은 N = 5인 지도와 계획이다

입력

첫째 줄에 공간의 크기를 나타내는 N이 주어집니다. (1<=N<=100)
둘째 줄에 여행가 A가 이동할 계획서 내용이 주어집니다. (1<=이동 횟수<=100)

출력

첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력

입력 예시
5
R R R U D D

출력 예시
3 4

풀이

접근 1

1. 일단... 간단하게 풀어보자 해서 들어올 때마다 LRUD를 분기처리 하자

Note

1. 스트링이면 data = input().split() 으로 해도 된다

제출 1 - 정답

n = int(input())
arr = list(map(str,input().split()))

now = [1,1]

for i in arr:
    if i=='L':
        if now[1]>1:
            now[1]-=1
    elif i=='R':
        if now[1]<n:
            now[1]+=1
    elif i=='U':
        if now[0]>1:
            now[0]-=1
    elif i=='D':
        if now[1]<n:
            now[0]+=1

print(now[0],now[1])

오답노트

책에 있는 풀이 진짜 약간 너무... 똑똑해서.. 이번엔 진짜 이건 못했을 것 같다구 생각함 ... 진짜 이렇게 문제 풀 때 저렇게 배열로 저장해서 이동하는 거 좀 더 잘 할 수 있도록 풀 때마다 신경을 써보면 좋을 것 같다

• 책에 있는 솔루션

# 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)

결과

• 풀이시간 : 8분

References

이것이 코딩 테스트다 with 파이썬 - 나동빈 저