该题如果空间复杂度为O(n)的话就比较简单,但是如果要使空间复杂度为O(1)则有如下两种方法。
方法一
数组为[1,2,3,4,5,6,7],数组长度ll为7,k为2
观察移动过后,假设将数组分为[0,1]部分1和[2,7]部分2,则部分1会在部分2后面去。
所以先将数组整体翻转得到:[7,6,5,4,3,2,1]。然后再将[0,1]部分和[2,7]部分翻转,得到[6,7,1,2,3,4,5],也就是结果。
class Solution {
public static void rotate(int[] nums, int k) {
k = k % nums.length; // 这步不能省
if (nums.length == 0 || k == 0 || k == nums.length){
return;
}
// 整体颠倒
int low = 0;
int high = nums.length - 1;
while (low < high){
int temp = nums[low];
nums[low] = nums[high];
nums[high] = temp;
low++;
high--;
}
// 翻转前部分
low = 0;
high = k-1;
while (low < high){
int temp = nums[low];
nums[low] = nums[high];
nums[high] = temp;
low++;
high--;
}
// 翻转后部分
low = k;
high = nums.length-1;
while (low < high){
int temp = nums[low];
nums[low] = nums[high];
nums[high] = temp;
low++;
high--;
}
}
}
方法二
方法二是类似于一种循环赋值。
num[index] 赋值给 num[(index + k) % ll],然后index= (index + k) % ll。如此循环,直到所有的数组值都被赋值过。
如果数组的长度为奇数,则当index = 0时,就可以遍历完所有数组元素。但是如果数组的长度为偶数,则一次循环并不能遍历完所有数组元素。
此时可以用count来记录赋过值的数组元素的数量。当count达到数组长度以后,就可以停止循环。此时可以用一个for循环,从index=0开始,一直index++。
循环赋值的过程逻辑有些麻烦,但是可以慢慢摸索,举一个例子。
class Solution {
public static void rotate(int[] nums, int k) {
if (nums.length == 0 || k == 0 || k == nums.length){
return;
}
int ll = nums.length;
int visit = 0; // 目前访问过的数字的数目
for (int i = 0; visit < ll; i++){
int index = i;
int temp1 = nums[index]; // 第一个值
int temp2 = nums[(index + k) % ll]; // 第二个值
index = (index + k) % ll; // 此时index指向第二个值
do{
nums[index] = temp1; // 把第一个值赋给第二个值
visit++; // 访问数++
temp1 = temp2; // 往后走,第一个值变成第二个值
index = (index + k) % ll;
temp2 = nums[index]; // 第二个值变成其再往后第一个值
}while (index != i); // 当index=i时,说明一个循环结束
nums[index] = temp1; // 但是此时最初的一个值并没有被赋值,循环就结束了,所以while结束以后为其赋值
visit++;
}
}
}