[Java/알고리즘] Codility Lesson 2 Arrays : CyclicRotation

문제 설명

---

 

2. Arrays lesson - Learn to Code - Codility

Task description
An array A consisting of N integers is given. Rotation of the array means that each element is shifted right by one index, and the last element of the array is moved to the first place. For example, the rotation of array A = [3, 8, 9, 7, 6] is [6, 3, 8, 9, 7] (elements are shifted right by one index and 6 is moved to the first place).

The goal is to rotate array A K times; that is, each element of A will be shifted to the right K times.

Write a function:

class Solution { public int[] solution(int[] A, int K); }

that, given an array A consisting of N integers and an integer K, returns the array A rotated K times.

For example, given

    A = [3, 8, 9, 7, 6]
    K = 3
the function should return [9, 7, 6, 3, 8]. Three rotations were made:

    [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]
For another example, given

    A = [0, 0, 0]
    K = 1
the function should return [0, 0, 0]

Given

    A = [1, 2, 3, 4]
    K = 4
the function should return [1, 2, 3, 4]

Assume that:

N and K are integers within the range [0..100];
each element of array A is an integer within the range [−1,000..1,000].

 

배열을 회전 시키는 문제이다.

주어진 숫자에 따라서 가장 뒤 요소를 앞으로 옮기는 문제.

 

 

문제 풀이

---

import java.util.LinkedList;
import java.util.Queue;
class Solution {
    public int[] solution(int[] a, int k) {
        if(a.length == 0) {
            return a;
        }

        Queue<Integer> q = new LinkedList<>();
        for (int i = a.length; i > 0; i--) {
            q.offer(a[i - 1]);
        }

        for (int i = 0; i < k; i++) {
            q.offer(q.poll());
        }

        int index = a.length - 1;
        for (Integer integer : q) {
            a[index] = integer;
            index--;
        }

        return a;
    }
}

// deque 사용 풀이
import java.util.Deque;
public static int[] solution(int[] a, int k) {
        if(a.length == 0) {
            return a;
        }
        int[] result = new int [a.length];

        Deque<Integer> dq = new LinkedList<>();
        for (int i : a) {
            dq.offer(i);
        }

        for (int i = 0; i < k; i++) {
            dq.offerFirst(dq.pollLast());
        }

        int index = 0;
        for (Integer integer : dq) {
            result[index] = integer;
            index++;
        }
        
        return result;
    }
}

•  Queue에 뒤집어서 넣고 했지만... Deque를 썼다면 더 좋을듯하다.
• Codility 풀이 링크

 

다른 사람의 풀이

---

public int[] solution(int[] A, int K) {
    int length = A.length;
    int[] result = new int[length];
    for (int i = 0; i < length; i++) {
        int index = (K + i) % length;
        result[index] = A[i];
    }
    return result;
}

• 왜 이런 풀이가 안 떠오를까...