Codility Lesson 02 - CyclicRotation( JAVA )

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

CyclicRotation coding task - Learn to Code - Codility

문제

- 배열 A를 K번만큼 우측으로 shift했을 때, 배열을 구하는 문제

- 마지막 원소는 배열의 첫번째로 들어오는 방식

 

풀이

- java deque(데크)를 활용하는 방식으로 풀이

- A배열의 원소들을 deque에 입력

- K번 만큼 반복시키며 마지막 원소를 빼서 첫번째에 다시 삽입한 후 배열 return

 

소스코드

// you can also use imports, for example:
import java.util.*;

// you can write to stdout for debugging purposes, e.g.
// System.out.println("this is a debug message");

class Solution {
    public int[] solution(int[] A, int K) {
        // write your code in Java SE 8
        Deque<Integer> deque = new LinkedList<Integer>();

        for(int i:A){
            deque.add(i);
        }
        // 3 8 9 7 6
        for( int i=0; i<K; i++ ){
            int lastVal = deque.pollLast();
            deque.addFirst(lastVal);
        }

        Object[] arr = deque.toArray();
        int[] answer = new int[arr.length];
        for(int i=0; i<arr.length; i++ ){
            answer[i] = (int)arr[i];
        }

        return answer;
    }
}

 

마지막에 deque로 toArray() 메소드를 실행 시 Object[]로 나오는 부분을 더 매끄럽게 처리할 수 있는 방법을 찾아봐야겠다..

 

참고 사이트

https://hbase.tistory.com/128

[Java] Deque (덱/데크) 사용법 및 예제

 

empty Array에 대한 예외처리를 하지 않았음;; 코딜리티는 코너케이스를 항상 확인이 필요하다 ^^;