[프로그래머스 / Python] 배열 회전시키기

문제

입출력 예

 

 

 

def solution(numbers, direction) :
    
    if direction == 'right' :
        temp = numbers[-1]
        numbers[1:] = numbers[0:-1]
        numbers[0] = temp
        
    else :
        temp = numbers[0]
        numbers[0:-1] = numbers[1:]
        numbers[-1] = temp
    
    return numbers

 

정답 코드

 

'right'일 경우, 소실될 맨 마지막 요소[-1]를 temp 변수에 저장해둔 후 모든 요소를 오른쪽으로 밀어준다.

그리고 저장해둔 temp 값을 [0] 위치에 할당한다.

 

'left'는 right와 반대로 실시한다.