정수가 담긴 배열 numbers와
문자열 direction가 매개변수로 주어집니다.
배열 numbers의 원소를 direction방향으로
한 칸씩 회전시킨 배열을 return하도록 solution 함수를 완성해주세요.
내 코드
from collections import deque
def solution(numbers, direction):
answer = []
d = deque()
d.extend(numbers)
print(d)
if direction == "right":
element = d.pop()
d.appendleft(element)
elif direction == "left":
element = d.popleft()
d.append(element)
answer = list(d)
return answer
예시 코드
def solution(numbers, direction):
return [numbers[-1]] + numbers[:-1] if direction == 'right' else numbers[1:] + [numbers[0]]