Search

PGS 실습용 로봇

태그
구현

문제 설명

주어진 명령어 문자열을 따라 로봇이 2차원 좌표 평면 위에서 움직인 후 최종 위치의 좌표 [x,y][x, y]를 반환하는 문제

예제 입력/출력

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

제약 조건

1command1,000,0001 ≤ |command| ≤ 1,000,000

문제 풀이

풀이 브루트 포스 - 상한 O(1,000,000)O(1,000,000)

풀이 코드

def solution(command): d = [(0,1),(1,0),(0,-1),(-1,0)] d_idx = 0 answer = [0,0] for c in command: if c == 'R': d_idx = (d_idx+1)%4 elif c == 'L': d_idx = (d_idx-1)%4 elif c == 'G': answer[0]+=d[d_idx][0] answer[1]+=d[d_idx][1] else: answer[0]-=d[d_idx][0] answer[1]-=d[d_idx][1] return answer
Python
복사