问题:
Rotate an array of n elements to the right by k steps.
For example, with n = 7 and k = 3, the array [1,2,3,4,5,6,7] is rotated to [5,6,7,1,2,3,4].
大意:
旋转一个有n个元素的数组,右移k步。
比如,n = 7,k = 3,数组 [1,2,3,4,5,6,7] 就被旋转为 [5,6,7,1,2,3,4]。
思路:
旋转本身没太多特别好说的,我的做法是用另一个数组来记录旋转后的内容,然后复制回原数组。当然记录时是从第nums.length-k个元素开始记录,记录到末尾后再去从头开始记录到刚才那个元素。
需要注意的是这里并非返回一个数组,程序会直接读取原数组位置的内容来检查,所以需要把旋转后的结果一个个复制回去。
还有,给出的k可能会大于数组长度,这时候就要对原数组长度取模,才会得出真正需要旋转的个数。
代码(Java):
public class Solution {
public void rotate(int[] nums, int k) {
k %= nums.length;
int[] rotateNums = new int[nums.length];
int index = 0;
for (int i = nums.length-k; i < nums.length; i++) {
rotateNums[index] = nums[i];
index++;
}
for (int i = 0; i < nums.length-k; i++) {
rotateNums[index] = nums[i];
index++;
}
for (int i = 0; i < nums.length; i++) {
nums[i] = rotateNums[i];
}
return;
}
}
他山之石:
public void rotate(int[] nums, int k) {
if(nums == null || nums.length < 2){
return;
}
k = k % nums.length;
reverse(nums, 0, nums.length - k - 1);
reverse(nums, nums.length - k, nums.length - 1);
reverse(nums, 0, nums.length - 1);
}
private void reverse(int[] nums, int i, int j){
int tmp = 0;
while(i < j){
tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
i++;
j--;
}
}
这里他用了三次不同的翻转函数,先翻转第一到要旋转的元素之前的元素,然后翻转要旋转的那些元素,最后翻转所有元素。这么一个过程之后就神奇地得到了正确的结果。
合集:https://github.com/Cloudox/LeetCode-Record