배열 회전시키기

송용진·2023년 8월 25일
0

알고리즘

목록 보기
65/173

정수가 담긴 배열 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]]
profile
백엔드 개발자

0개의 댓글