面试经典 150 题 ---- 轮转数组

轮转数组

方法一：使用额外的数组

我们可以使用额外的数组来将每个元素放至正确的位置。用 n 表示数组的长度，我们遍历原数组，将原数组下标为 i 的元素放至新数组下标为 (i+k) mod n 的位置，最后将新数组拷贝至原数组即可。

class Solution {
    public void rotate(int[] nums, int k) {
        int len = nums.length;
        int[] newArr = Arrays.copyOf(nums, len);
        for (int i = 0; i < len; i ++ ) {
            nums[(i + k) % len] = newArr[i];
        }
    }
}

时间复杂度： O(n)
空间复杂度： O(n)

java 中拷贝数组的方法：

1. System.arraycopy() 方法:

System.arraycopy(源数组, 源数组起始位置, 目标数组, 目标数组起始位置, 拷贝长度);

int[] sourceArray = {1, 2, 3, 4, 5};
int[] destinationArray = new int[5];
System.arraycopy(sourceArray, 0, destinationArray, 0, sourceArray.length);

2. Arrays.copyOf() 方法:

int[] newArray = Arrays.copyOf(源数组, 新数组长度);

int[] sourceArray = {1, 2, 3, 4, 5};
int[] newArray = Arrays.copyOf(sourceArray, sourceArray.length);

3. Arrays.copyOfRange() 方法:

int[] newArray = Arrays.copyOfRange(源数组, 起始位置, 结束位置);

int[] sourceArray = {1, 2, 3, 4, 5};
int[] newArray = Arrays.copyOfRange(sourceArray, 0, sourceArray.length);

4. 使用clone() 方法:

int[] newArray = sourceArray.clone();

int[] sourceArray = {1, 2, 3, 4, 5};
int[] newArray = sourceArray.clone();

方法二：数组翻转

我们将数组向右移动 k 个位置，那么数组尾部的 k % n 个元素就会移动到数组的头部，同样数组中其它元素就会向后移动 k % n 个位置。
我可以将整个数组进行一次翻转，这样尾部的 k % n 个数组就会移动到头部，然后再分别翻转 [0, k % n - 1] 部分的数组和 [k % n, n - 1] 部分的数组就可以得到答案。

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

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

时间复杂度： O(n)
其中 n 为数组的长度。每个元素被翻转两次，一共 n 个元素，因此总时间复杂度为 O(2n)=O(n)

空间复杂度： O(1)