LeetCode-轮转数组

题目描述

给定一个整数数组 nums，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右轮转 1 步: [7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]

示例 2:

输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]
解释: 
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

思路解析

本题的含义很清晰，本质就是要在原位置移动数组，这里可以有两种思路。

首先我们可以借助另外一个数组储存移动过后的数组，最后再复制到原数组之中，这种思路很好理解，不过需要借助另外的数组，在时间和空间上的速度就略显不足了。

具体代码如下：

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        int[] newArr = new int[n];
        for (int i = 0; i < n; ++i) {
            newArr[(i + k) % n] = nums[i];
        }
        System.arraycopy(newArr, 0, nums, 0, n);
    }
}

另外一种思路是通过三次翻转的方式，在原数组中即可完成移动。具体做法为：
先把数组整体反转

再把前k个和后几个再次反转，这样既移动了位置又把顺序恢复了回来，很巧妙。

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        k %= n; 
        reverse(nums, 0, n - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, n - 1);
    }

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