(코딜리티) Lesson 2 : CyclicRotation - Python

풀면서 읭? 이런건 너무 쉬운데? 라는 생각으로 어이없게 풀었네요..

물론 제한사항을 제대로 안봐서 첫번째엔 효율성에서 87%가 나왔지만요 ㅎㅎ...

 

문제 : app.codility.com/programmers/lessons/2-arrays/cyclic_rotation/

CyclicRotation coding task - Learn to Code - Codility

문제 설명

N개의 Array인 A와 정수인 K가 주어졌을 때, 오른쪽으로 하나씩 움직이고 마지막 요소는 제일 처음으로 옮기시오.

 

예를 들어, A = [3, 8, 9, 7, 6]인경우 K가 1이면 [6, 3, 8, 9, 7]을 return합니다.

 

제한 사항

• N과 K는 0~100사이의 정수입니다.
• A의 각 요소는 -1000~1000사이의 정수입니다.

코드

def solution(A, K):
    if A :
    	for i in range(K):
            a = A.pop()
            A.insert(0, a)

    return A