[PCCP 모의고사 2] 1번 - Python

https://school.programmers.co.kr/learn/courses/15009/15009-pccp-%EB%AA%A8%EC%9D%98%EA%B3%A0%EC%82%AC-2%ED%9A%8C

PCCP 모의고사 2회

 

1번 문제는 전형적인 구현문제였다. 예전에 했던 것을 기억하여 그냥 그대로 구현하였더니 쉽게 풀렸다.

 

def solution(command):
    cur = [0,0]
    direction = 0
    directions = [(0,1),(1,0),(0,-1),(-1,0)] # 위 오 아 왼
    for i in command:
        if i == 'R':
            direction = (direction+1)%4
        elif i == 'L':
            direction = (direction+-1)%4
        elif i == 'G':
            cur[0] += directions[direction][0]
            cur[1] += directions[direction][1]
        elif i == 'B':
            cur[0] -= directions[direction][0]
            cur[1] -= directions[direction][1]
    return cur