프로그래머스-배열 회전시키기(파이썬, python)

SA Jung·2022년 10월 5일
0

Programmers 문제 풀이

목록 보기
10/14

문제링크

배열 회전시키기
문제 설명
정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다. 배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을 return하도록 solution 함수를 완성해주세요.

제한사항
1 ≤ numbers의 길이 ≤ 100
direction은 "left" 와 "right" 둘 중 하나입니다.

입출력 예 설명
입출력 예 #1

numbers 가 [1, 2, 3]이고 direction이 "right" 이므로 오른쪽으로 한 칸씩 회전시킨 [3, 1, 2]를 return합니다.
입출력 예 #2

numbers 가 [4, 455, 6, 4, -1, 45, 6]이고 direction이 "left" 이므로 왼쪽으로 한 칸씩 회전시킨 [455, 6, 4, -1, 45, 6, 4]를 return합니다.

1. 풀이

def solution(numbers, direction):
    from collections import deque
    numbers_deque = deque(numbers)
    if direction == "right":
        numbers_deque.rotate(1)
    elif direction == "left":
        numbers_deque.rotate(-1)
    return list(numbers_deque)
  • list를 deque형식으로 만들어서 deque에서 제공하는 함수를 활용하면 아주 쉽게 풀 수 있다.

  • "배열 회전시키기"의 출제 목표가 deque.rotate를 알고 있느냐? 활용할 수 있느냐?가 핵심인 것 같다.

  • deque.rotate(양수의 숫자) 이면 시계방향으로 숫자만큼 돈다. 즉, 오른쪽으로 숫자만큼 회전한다.

  • deque.rotate(음수의 숫자) 이면 반시계방향으로 숫자만큼 돈다. 즉, 왼쪽으로 숫자만큼 회전한다.

참조

profile
Tomorrow will be better than yesterday :)

0개의 댓글