본문 바로가기

Algorithm/Implementation

(Python3) - 프로그래머스(코딩테스트 입문) : 배열 회전시키기

반응형

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

간단 구현 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

정답 변수 answer 선언 후 빈 배열로 초기화합니다.

📔 풀이과정

1. direction이 right인 경우: 마지막 원소가 배열의 첫 번째 원소가 되어야 하므로 answer에 마지막 원소를 append한 뒤 마지막 원소를 제외한 sliced된 numbers를 extend해줍니다.

 

2. left인 경우 : 첫 번째 원소가 배열의 마지막 번째 원소가 되어야 하므로 answer에 첫 번째 원소를 제외한 sliced된 numbers로 갱신하고 뒤에 첫 번째 원소를 추가해줍니다.

📔 정답 출력 | 반환

answer를 반환합니다.


📕 Code

📔 Python3

def solution(numbers, direction):
    answer = []
    if direction == 'right':
        answer.append(numbers[-1])
        answer.extend(numbers[:-1])
    else:
        answer = numbers[1:]
        answer.append(numbers[0])
    return answer

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.