题目描述:
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: [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:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
解题思路:
思路一:环状替代【容易想到,不容易理清】
用下面的例子说明 环状替代 过程:
nums = [1, 2, 3, 4, 5, 6]
k = 2
对于一个长度为 n 的数组,整体移动 k 个位置
- 当 n 和 k 的最大公约数 等于 1 的时候:1 次遍历就可以完成交换;比如 n = 5, k = 3
- 当 n 和 k 的最大公约数 不等于 1 的时候:1 次遍历是无法完成的所有元素归位,需要 m (最大公约数) 次
所以在最大公约数不为 1 的时候
比如 [A,B,C,D,E,F] 此时 n=6 ,k=4 ,其最大公约数为 2 ,因此需要 2 轮循环
我们就可以把这个数组分成两部分来看:
第 1 轮循环(分组1):A E C [A]
第 2 轮循环(分组2):B F D [B]
即:每一轮循环只会在自己的那一组上不停的遍历。所以
数组的前 m 个元素,其实就是每一个分组的第一个元素,我们控制流程在每次发现一轮循环走到原点时+1
如何判断所有的分组都执行归位了呢? 可以有两种方法来控制
第一种:我们就用最大公约数 m 来控制外循环,代表总共有 m 轮循环;
第二种:由于n个元素归位需要n次交换,所以我们定义一个count代表交换次数,当 count = n 时完成;
class Solution {
public void rotate(int[] nums, int k) {
k=k%nums.length;
int count = 0; //移动总次数,总次数需要
//k=3,nums.length 为偶数时,会形成多个换环 1 2 3 4 5 6(1 3 1,2 4 2,3,1,3) ,为奇数时则只有一个环 1 2 3 4 5 6 7 (1 4 7 3 6 2 5 1),i=0不会增加
for (int start = 0; count < nums.length; ++start) {
int current = start; //当前坐标
int prev = nums[start]; //前一个坐标
do {
int next = (current + k) % nums.length; //下一个坐标
//交换
int temp = nums[next];
nums[next] = prev;
prev = temp;
current = next;
count++;
} while (start != current); // 循环暂停,回到起始位置
}
}
}
思路二:数组翻转【不容易想到,容易理清】
该方法基于如下的事实:当我们将数组的元素向右移动 k 次后,尾部(k mod n)个元素会移动至数组头部,其余元素向后移动(k mod n)个位置。
该方法为数组的翻转:我们可以先将所有元素翻转,这样尾部的(k mod n)个元素就被移至数组头部,然后我们再翻转 [0, (k mod n)−1] 区间的元素和 [(k mod n),n−1] 区间的元素即能得到最后的答案。
我们以 n=7n=7,k=3k=3 为例进行如下展示:
class Solution {
public void rotate(int[] nums, int k) {
k %= nums.length;
reverse(nums, 0, nums.length - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, nums.length - 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;
}
}
}