Skip to content

Rotate array

1. Problem

해당 문제는 여기에서 확인하실 수 있습니다.

2. Answer

def solution(numbers, direction):
    # 만약 방향이 "right"이면
    if direction == "right":
        # 마지막 원소를 맨 앞으로 이동시키고 나머지 원소들을 뒤에 붙임
        return [numbers[-1]] + numbers[:-1]
    # 만약 방향이 "left"이면
    elif direction == "left":
        # 첫 번째 원소를 맨 뒤로 이동시키고 나머지 원소들을 앞에 붙임
        return numbers[1:] + [numbers[0]]