[Cos Pro 1급, Python] 2차 6번 : 로봇을 움직여주세요

문제

로봇이 아래 그림과 같이 2차원 평면의 원점 (0, 0)에 서있습니다.

이 로봇은 x축 방향, 혹은 y축 방향으로만 움직일 수 있으며, 알파벳으로 명령을 내릴 수 있습니다. 명령을 내릴 때 사용하는 알파벳은 'L', 'R', 'U', 'D'의 4가지이며, 'L'은 x축 방향으로 -1만큼, 'R'은 x축 방향으로 +1만큼, 'U'는 y축 방향으로 +1만큼, 'D'는 y축 방향으로 -1 만큼 이동하라는 의미입니다.

로봇에게 내린 명령이 순서대로 들어있는 문자열 commands가 매개변수로 주어질 때, 주어진 명령을 모두 수행한 후의 로봇 위치를 return 하도록 solution 함수를 완성해주세요.

 

 

 

 

 

 

 

코드

def solution(commands):
	answer = []
	ans = [0, 0]
	answer = ans
	x = 0
	y = 0
	
	for i in commands :
		if i == 'R' :
			x += 1
		elif i == 'D' :
			y -= 1
		elif i == 'L' :
			x -= 1
		else :
			y += 1
			
			
	answer[0] = x
	answer[1] = y
	return answer
    
    
    
    
commands = "URDDL"
ret = solution(commands)

print("solution 함수의 반환 값은", ret, "입니다.")