[PCCP 모의고사#2] 실습용 로봇 (단순 구현)

문제 설명

컴퓨터공학과에서는 실습용 로봇을 이용해서 로봇 프로그래밍을 학습합니다. 실습용 로봇은 입력된 명령에 따라 x좌표와 y좌표로 표현되는 2차원 좌표 평면 위를 이동합니다. 하나의 명령은 하나의 문자로 주어지며 각 명령어에 따라 로봇이 수행하는 일은 다음과 같이 네 종류입니다.

• 'R': 로봇이 오른쪽으로 90도 회전합니다.
• 'L': 로봇이 왼쪽으로 90도 회전합니다.
• 'G': 로봇이 한 칸 전진합니다.
• 'B': 로봇이 한 칸 후진합니다.

명령어는 각각의 명령들이
모인 하나의 문자열로 주어지며, 차례대로 수행됩니다.

로봇은 처음에 (0, 0) 위치에 +y 축을 향하여 놓여 있습니다.

다음 그림은 번호 순서대로 명령어 "GRGLGRG"의 과정을 보여줍니다.

 

 

 

 

 

 

 

로봇에 입력된 명령어를 순서대로 담고 있는 문자열 command가 주어집니다. 로봇이 주어진 명령어들을 순서대로 모두 수행한 뒤 도착한 최종 위치의 좌푯값 x, y를 순서대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

제한 사항

• 1 ≤ commad의 길이 ≤ 1,000,000
• command는 'R', 'L', 'G', 'B'으로 구성된 문자열입니다.
• command에 들어있는 문자 하나하나가 각 명령을 나타내며, 문자열에 먼저 등장하는 명령을 먼저 수행해야 합니다.

입출력 예

command	result
"GRGLGRG"	[2, 2]
"GRGRGRB"	[2, 0]

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 로봇이 이동한 좌표는 (0, 0) → (0, 1) → (1, 1) → (1, 0) → (2, 0) 입니다.

풀이

• 이거 게임 만들 때 많이 쓰던 방식이라 쉬웠다.
  • 슈팅게임에서 4방향, 8방향, 16방향, 32방향 등등할 때 주로 사용하는 방식
• R/L입력하면 이동 단위 배열 변경
  • 초기상태 : [1,0]
  • RRR : [0,1],[-1,0],[0,-1]
  • R은 배열+1
  • L은 배열-1
  • 배열 넘어가면 반대쪽으로 가기
• G/B입력하면 이동 단위 만큼 현재 좌표에 더해주기
  • 끝? 무지성 코딩 가즈아!
    • 어? 왜 실패?

테스트 1
입력값 〉	"GRGLGRG"
기댓값 〉	[2, 2]
실행 결과 〉	테스트를 통과하였습니다.
출력 〉	
G
[1, 0] 0
R
[1, 0] 1
G
[1, 1] 1
L
[1, 1] 0
G
[2, 1] 0
R
[2, 1] 1
G
[2, 2] 1

테스트 2
입력값 〉	"GRGRGRB"
기댓값 〉	[2, 0]
실행 결과 〉	실행한 결괏값 [0,2]이 기댓값 [2,0]과 다릅니다.
출력 〉	
G
[1, 0] 0
R
[1, 0] 1
G
[1, 1] 1
R
[1, 1] 2
G
[0, 1] 2
R
[0, 1] 3
B
[0, 2] 3

• 아놔... 이번에도 행렬 바꿔놨네 ㅋㅋㅋ
  • 테스트1은 통과하는데, 테스트2는 통과를 못 해서 보니까...
    • 좌표를 또 반대로 해놔서, 코드 고치긴 싫고 마지막에 answer만 뒤집어 줌

def solution(command):
    answer = [0,0]
    cmd_mapping = [[1,0],[0,1],[-1,0],[0,-1]]
    robot_dir = 0
    for c in command:
        #print(c)
        if c == "R":
            robot_dir += 1
            robot_dir %= 4
        elif c == "L":
            robot_dir -= 1
            if robot_dir < 0: robot_dir = 3
        elif c == "G":
            answer = [ai + ri for ai, ri in zip(answer,cmd_mapping[robot_dir])]
        elif c == "B":
            answer = [ai - ri for ai, ri in zip(answer,cmd_mapping[robot_dir])]
        #print(answer, robot_dir)
    answer[0],answer[1] = answer[1],answer[0]
    return answer

정확성  테스트
테스트 1 〉	통과 (275.99ms, 11.6MB)
테스트 2 〉	통과 (210.95ms, 11.4MB)
테스트 3 〉	통과 (110.56ms, 10.5MB)
테스트 4 〉	통과 (165.47ms, 11.1MB)
테스트 5 〉	통과 (221.77ms, 11.4MB)
테스트 6 〉	통과 (436.03ms, 11.8MB)
테스트 7 〉	통과 (287.71ms, 11.9MB)
테스트 8 〉	통과 (274.41ms, 11.9MB)
테스트 9 〉	통과 (297.49ms, 12MB)
테스트 10 〉	통과 (359.95ms, 11.9MB)

• 고수의 코드
  • 깔끔하네...

def solution(command):
    path = [[0,1], [1,0], [0,-1], [-1, 0]] 
    x = y = d = 0

    for i in command:
        if i == 'R': d = (d+1)%4
        elif i == 'L': d = (d+3)%4
        elif i == 'G':
            x += path[d][0]
            y += path[d][1]
        elif i == 'B':
            x -= path[d][0]
            y -= path[d][1]

    return [x,y]