Description

Given an array, rotate the array to the right by k steps, where k is non-negative.

Example 1:

Input: nums = [1,2,3,4,5,6,7], k = 3
Output: [5,6,7,1,2,3,4]
Explanation:
rotate 1 steps to the right: [7,1,2,3,4,5,6]
rotate 2 steps to the right: [6,7,1,2,3,4,5]
rotate 3 steps to the right: [5,6,7,1,2,3,4]

Example 2:

Input: nums = [-1,-100,3,99], k = 2
Output: [3,99,-1,-100]
Explanation:
rotate 1 steps to the right: [99,-1,-100,3]
rotate 2 steps to the right: [3,99,-1,-100]

Constraints:

• 1 <= nums.length <= 105
• 231 <= nums[i] <= 231 - 1
• 0 <= k <= 105

Follow up:

• Try to come up with as many solutions as you can. There are at least three different ways to solve this problem.
• Could you do it in-place with O(1) extra space?

Solution

• 힌트 1에서 additional memory를 쓰는 게 제일 쉬운 방법이래서 일단 그렇게 해봤다.
  • 우선 rotate할 횟수인 k를 배열 사이즈로 나눈 나머지값으로 설정
  • 새로운 배열을 만들고 인덱스 값에 k를 더했을 때 배열의 사이즈를 넘어가면 배열의 사이즈만큼 빼준다 → 배열의 앞자리로 들어가게 됨

// using additional memory
public void rotate(int[] nums, int k) {
        int[] temp = new int[nums.length];
        k %= nums.length;
        int i;
        for (i = 0; i < nums.length; i++) {
            if (i + k >= nums.length) {
                temp[i + k - nums.length] = nums[i];
            } else temp[i + k] = nums[i];
        }

        for (i = 0; i < nums.length; i++) {
            nums[i] = temp[i];
        }

        for (int j : nums) {
            System.out.println(j);
        }
}

• 근데 사실 이러라고 있는 미디움 문제가 아니다. 힌트에서는 배열을 reverse sort 해보라고 했다. 배열을 전체적으로 reverse하면 인덱스 값으로 k - 1을 기준으로 배열을 둘로 나눠 그 안에서 다시 reverse sort를 해주면 된다. 머리로는 안 보이는데 적어가면서 하니까 잘 보였당..

// using two pointer
    public void rotate2(int[] nums, int k) {
        k %= nums.length;

        reverse(nums, 0, nums.length - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, nums.length - 1);
    }

    private void reverse(int[] nums, int start, int end) {
        int i = start, j = end;
        while (i < j) {
            int temp = nums[i];
            nums[i] = nums[j];
            nums[j] = temp;
            i++;
            j--;
        }
    }