思路
解法1
这道题最简单直接的方式用一个双端队列,从队尾依次取出k个元素插入的队头。
代码如下:
class Solution {
public void rotate(int[] nums, int k) {
if (nums == null || nums.length == 0 || k < 0) {
throw new IllegalArgumentException("Illegal argument!");
}
int actualRotations = k % nums.length;
Deque<Integer> deque = new ArrayDeque<>();
// 将数组元素添加到双端队列中
for (int num : nums) {
deque.addLast(num);
}
// 进行旋转
for (int i = 0; i < actualRotations; i++) {
int lastElement = deque.removeLast();
deque.addFirst(lastElement);
}
// 将旋转后的元素放回数组
int idx = 0;
for (int num : deque) {
nums[idx++] = num;
}
}
}
解法2
将原数组的后k
个元素先放入新数组的前k
个位置,然后将原数组剩余的元素依次放入新数组的后续位置,最后再将新数组的元素复制回原数组。
class Solution {
public void rotate(int[] nums, int k) {
if (nums == null || nums.length == 0) {
return;
}
int N=nums.length;
// 如果 k 大于数组长度,则进行取模运算
k = k % N;
int[] arr = new int[N];
for (int i = 0; i <k ; i++) {
arr[i]=nums[N-k+i];
}
for (int i = 0; i < N - k; i++) {
arr[k+i]=nums[i];
}
for (int i = 0; i < N; i++) {
nums[i]=arr[i];
}
}
}