面试经典 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 中拷贝数组的方法：
- System.arraycopy() 方法:
System.arraycopy(源数组, 源数组起始位置, 目标数组, 目标数组起始位置, 拷贝长度);
int[] sourceArray = {1, 2, 3, 4, 5};
int[] destinationArray = new int[5];
System.arraycopy(sourceArray, 0, destinationArray, 0, sourceArray.length);
- Arrays.copyOf() 方法:
int[] newArray = Arrays.copyOf(源数组, 新数组长度);
int[] sourceArray = {1, 2, 3, 4, 5};
int[] newArray = Arrays.copyOf(sourceArray, sourceArray.length);
- Arrays.copyOfRange() 方法:
int[] newArray = Arrays.copyOfRange(源数组, 起始位置, 结束位置);
int[] sourceArray = {1, 2, 3, 4, 5};
int[] newArray = Arrays.copyOfRange(sourceArray, 0, sourceArray.length);
- 使用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)
