[이것이 코딩테스트다] 상하좌우 (Python)

문제

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

---

 

입력 조건

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

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

 

출력 조건

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

 

---

입력 예시                                                                            출력 예시

5

R R R U D D                                                                        3 4

 

---

좌표 문제는 조금 풀어봐서 어떻게 해야 할지 대충 생각을 하고 있었는데

그냥 냅다 If문 써버렸다

 

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

n = int(input())
arr = list(input().split())

now = [1,1]

for i in range(len(arr)):
    if arr[i] == 'R':
        if now[1] == n :
            continue
        else :
            now[1] +=1
            continue
    elif arr[i] == 'L':
        if now[1] == 1:
            continue
        else :
            now[1] -=1
            continue
    elif arr[i] == 'U' : 
        if now[0] == 1:
            continue
        else :
            now[0] -=1
            continue
    else :
        if now[0] == n:
            continue
        else:
            now[0] +=1
            continue

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

# ㅋㅋ 장난? 이딴식으로 짜지 마세요. 네.

좌표 문제에서 dx, dy 썼던 거 기억나서 일단 써두고 시작했는데 막상 코드에 적용을 안했다. 

다 풀고 답을 보니 어떻게 하는 거 였는지 기억이 나버렸당.ㅋㅋ

 

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

n = int(input())
arr = list(input().split())

move = ['R','L','D','U']
x,y = 1,1

for i in arr :
    for j in range(len(move)):
        if i == move[j] :
            nx = x + dx[j]
            ny = y + dy[j]
    if nx<1 or ny <1 or nx >n or ny>n:
        continue
    x = nx
    y = ny

print(y,x)

범위를 먼저 보고 더할지 스킵할지 고민하려 했으나

일단 다른 변수를 만들어 거기에 더해두고 범위를 넘어가면 실제 변수에는 적용을 안하는 방법을 이용했다.

다시 감 잡을 때 까지 화이팅~~~~~~~~~~~~~~~~~~~~~~~~~~~