题目来自leetcode网站:https://leetcode-cn.com/problems/rotate-array/
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
我的思路:
直接的想法是新建一个数组,往里面存数据,但是空间复杂度是On,目标是O1。通过观察一步一步的操作,可以发现,每一步的操作是一样的:把数组前n-1个元素向后移动一位,把第n个元素拿到数组头部。可以使用一个变量tmp来记录原有位置的值,然后类似于冒泡排序那样,做元素的两两交换位置,这样就是O1的空间复杂度了。
// int a;
// for(int n=1;n<=k;n++){
// a=nums[0];
// nums[0]=nums[nums.length-1];
// for(int i=1;i<nums.length-1;i++){
// //把当前位置记下来
// int tmp=nums[i];
// //把之前记得数给当前位置
// if(i==1){
// nums[i]=a;
// }else{
// nums[i]=tmp;
// }
// }
// }
不过一开始没有想到两两交换,于是我写了如上代码,先把下标0记下来,然后把第n个元素给下标0的位置,然后想着用循环,再把下标1记下来,把之前下标0的数给下标1的位置,但是没有写出来,总是哪里不对(因为先要纪录当前元素,然后把之前纪录的值赋值给当前位置,但是纪录当前元素的时候就把上一个元素的纪录给覆盖了,肯定是不行的,总之没写出来)。然后我想换一种思路:
//换一个思路:纪录最后一个之后,就是从后往前的两两交换
//int last;
for(int n=1;n<=k;n++){
//last=nums[nums.length-1];
for(int i=nums.length-1;i>0;i--){
int tmp=nums[i];
nums[i]=nums[i-1];
nums[i-1]=tmp;
}
//nums[0]=last;
}
从后往前两两交换之后,神奇的发现,直接就得到了第一步的结果了,很类似于冒泡,就是第一个是最大的值,把他交换到最后去了!
观察很重要!