[python3] 이것이 취업을 위한 코딩 테스트다 - 상하좌우 (ch4. 구현)

내돈내산

참고) 이것이 취업을 위한 코딩테스트다 with 파이썬 책을 기반으로 작성된 문제와 코드입니다. 
따라서 문제는 자세하게 적지 않고, 간단한 설명과 제 코드만 올리겠습니다.

http://www.yes24.com/Product/Goods/91433923

이것이 취업을 위한 코딩 테스트다 with 파이썬 - YES24

 

예제 4-1) 상하좌우

• 시뮬레이션 문제의 대표적인 형태
• 대체로 코테의 1~2번 문제는 그리디나 이런 구현 문제라고 한다. 

N x N 크기의 정사각형 공간이 있다. 
이동 계획을 입력받는다. ( L / R / U / D )

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

처음은 (1,1) 자리에서 시작하고, 이동 계획에 따라 이동한 후의 위치를 구하는 문제 
단, 좌표를 벗어나게 되는 이동계획은 무시된다.

입력과 출력의 예시는 다음과 같다.

입력예시
5
R R R U D D

출력예시
3 4

<문제 풀이>

나는 우선 첫 위치 (1,1)을 place로 지정했다. 

그리고 함수 속에 LRUD라는 이동 계획을 담은 함수를 따로 만들어서 진행했다.

<내가 푼 문제>

def solution(N, route):
    place = [1,1]
    
    def LRUD(a, place):
        temp_place = list(place)
        if a == 'R':
            temp_place[1] += 1
        elif a == 'L':
            temp_place[1] -= 1
        elif a == 'U':
            temp_place[0] -= 1
        elif a == 'D':
            temp_place[0] += 1

        if (temp_place[0] < 1) or (temp_place[1] < 1) or (temp_place[1] > N) or (temp_place[0] > N) :
            return place #위치가 벗어나면 가지않고 다시 place로 반환
        else: # 벗어나지 않는다면
            place = list(temp_place) # 변한걸로 업데이트 쳐주고
            return place
    
    for i in range(0, len(route)):
        a = route.pop(0)
        place = LRUD(a, place)
    return place

여기서 조금 시간이 걸렸던게, list를 복사하는 과정에서 자꾸 해결하지 못해서 인터넷 검색을 했다. ㅋㅋ....
리스트를 복사할때는 아래와 같은 사항을 주의해서 복사해야 함을 알았다!!

역시 날 괴롭히는건 기본 문법이야... 역시,,, 언제나 기초가 제일 중요해ㅜㅜ 

temp_list = list
# 이렇게 하면, 리스트 자체의 동일한 객체에 이름이 두개 생긴다고 이해하면 된다. 
# 따라서 list가 변해도 동시에 temp_list가 변한다. 

#그래서 리스트를 복사하고 싶다면, 
# 아래 처럼 다시 list 객체를 새롭게 불러와야한다. 
temp_list = list(list)

 

답안 출력)

N = 5
route = ['R','R','R','U','D','D']
solution(N, route)

 

 

<책에 있는 정답과 비교>

n = 5
x, y = 1, 1
plans = ['R','R','R','U','D','D']

# LRUD 에 대한 이동 방향을 4차원으로 표현
# 인덱스 0이 L, 1이 R, 2가 U ... 
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 # 더이상 해당 안의 for 문에서 돌지 않고 나감
    x, y = nx, ny

print(x,y)

책은 이렇게 문제를 풀어놨다. 내가 생각하지 못했던 방법으로
L R U D를 하나하나 입력받지 않고, dx, dy, move_types 의 인덱스로 설정할 수 있다는 것(?)을 배웠다.

아무래도 책에 나온 코드가 좀더 간결하고 짧기때문에 구현하는데 시간이 덜 걸릴 것 같다.