题目描述
【leetcode】189. 旋转数组( Rotate Array )
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
第一次解答
思路:
题意是把数组nums向右平移k次,超出边界的部分从数组首插入。
简单的方法是按部就班的平移1次,2次,…,一直到k次,该方法缺点就是每次平移所有元素都要移动一次,复杂度是k*nums.size()。
另一种思路是一步计算得到每个元素最终所在的位置,将元素直接放到最终位置。用一个临时变量保存原位置值,但是这个代码不好实现,我写了几次都有bug,最后还是参考了官方题解。
test case:
[]
1
[1]
1
[1,2]
1
[1,2]
2
[1,2]
3
[1,2,3,4,5,6,7]
3
[1,2,3,4,5,6]
2
class Solution {
public:
void rotate(vector<int>& nums, int k) {
if(nums.size() == 0)
return;
k = k%nums.size(); // 若k>=nums.size(),平移k次与平移k%nums.size()次效果相同
if(0 == k)
return;
int count = 0; // 记录已变换位置的元素个数
// 注意这个for循环执行的判断条件
// 若 nums.size() % k == 0, 则for循环执行k次,直到count==nums.size()
// 若 nums.size() % k != 0, 则for循环执行1次(因为在do-while里一直执行到count==nums.size())。
for(int start=0; count<nums.size(); ++start){
int current = start;
int last_value = nums[current];
int next;
// 注意这里用do-while而不是wihle
do{
next = (current + k) % nums.size();
int temp = nums[next];
nums[next] = last_value;
last_value = temp;
current = next;
++count;
}while(start != current);
}
}
};
结果:
截图