[프로그래머스][파이썬] 배열 회전시키기 - 배열 (Level 0)

◽ 문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/120844

◽ 문제

◽ 입력 & 출력

◽ 내 풀이

def solution(numbers, direction):
    answer = [0] * len(numbers)
    
    for i in range(len(numbers)):
        if direction == 'right':
            answer[(i + 1) % len(numbers)] = numbers[i]
        else:
            answer[(i - 1) % len(numbers)] = numbers[i]
            
    return answer

• 처음엔 반복문에서 스왑을 해볼까도 하고 고생 정말 많이 한 문제다.
• right 이면 현재 인덱스의 값을 다음 인덱스의 값에 저장하되, 리스트의 길이 나머지 연산을 통해 순환을 고려했다.
• 참고) 리스트의 길이로 나머지 연산하면 리스트 순환을 고려할 수 있다. https://velog.io/@pric/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EA%B3%B5-%EB%8D%98%EC%A7%80%EA%B8%B0-%EC%88%98%ED%95%99-Level-0
  
  

◽ 다른 사람 풀이

from collections import deque

def solution(numbers, direction):
    numbers = deque(numbers)
    
    if direction == 'right':
        numbers.rotate(1)
    else:
        numbers.rotate(-1)
        
    return list(numbers)

• 파이썬 내장 deque을 통해 똑똑하게 풀었다.
• deque이 어떤 자료구조인지는 알고 있었으나, rotate() 함수가 있는지는 몰랐다.
  
  

◽ 더 나아가기

• 파이썬의 deque 은 양쪽 끝에서 삽입과 삭제가 빠른 자료구조이다.
• 양쪽에서 모두 O(1) 연산이 가능하고, 연결 리스트 구조로 메모리 할당이 효율적이다.
• rotate() 함수는 인자가 양수라면 오른쪽으로, 음수라면 왼쪽으로 이동한다.
  
  
  
  

피드백은 언제나 환영입니다 :)