[Codility] [Lesson2] CyclicRotation JAVA

문제

• Link: app.codility.com/programmers/lessons/2-arrays/cyclic_rotation/

배열을 오른쪽으로 N번 이동하기

ex) 

A = [3, 8, 9, 7, 6]

K = 3

가 주어졌을때

 

[3, 8, 9, 7, 6] -> [6, 3, 8, 9, 7]

[6, 3, 8, 9, 7] -> [7, 6, 3, 8, 9]

[7, 6, 3, 8, 9] -> [9, 7, 6, 3, 8]

 

3번 로테이션 돌렸으니까 결과는 [9, 7, 6, 3, 8] 이다.

 

풀이

public int[] solution(int[] A, int K) {
    if (A == null || A.length == 0) {
        return A;
    }
    int tempNum = 0;
    for (int i = 0; i < K; i++) {
        tempNum = A[0];
        A[0] = A[A.length - 1];
        for (int j = A.length - 1; j > 0; j--) {
            if (j == 1) {
                A[j] = tempNum;
                continue;
            }
            A[j] = A[j-1];
        }
    }
    return A;
}

 

테스트

- 주어진 값

int[] A = {3, 8, 9, 7, 6};
int K = 3;

 

- 결과 값

[9, 7, 6, 3, 8]