给定一个数组,将数组中的元素向右移动 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) 的原地算法。
//章节 - 数组和字符串 //五、小结 //1.旋转数组 /* 算法思想: 一种O(n)的空间复杂度的方法,复制一个和nums一样的数组,然后利用映射关系i -> (i+k)%n来交换数字。 */ //算法实现: /* class Solution { public: void rotate(vector<int>& nums, int k) { vector<int> t = nums; for (int i = 0; i < nums.size(); ++i) { nums[(i + k) % nums.size()] = t[i]; } } }; */ /* 算法思想: 类似翻转字符的方法,思路是先把前n-k个数字翻转一下,再把后k个数字翻转一下,最后再把整个数组翻转一下。 */ //算法实现: /* class Solution { public: void rotate(vector<int>& nums, int k) { if (nums.empty() || (k %= nums.size()) == 0) return; int n = nums.size(); reverse(nums.begin(), nums.begin() + n - k); reverse(nums.begin() + n - k, nums.end()); reverse(nums.begin(), nums.end()); } }; */ /* 算法思想: 旋转数组的操作也可以看做从数组的末尾取k个数组放入数组的开头,通俗的讲,就是把前n-k个数依次加到数组末尾。所以我们用STL的push_back和erase可以很容易的实现这些操作。 */ //算法实现: class Solution { public: void rotate(vector<int>& nums, int k) { if (nums.empty() || (k %= nums.size()) == 0) return; int n = nums.size(); for (int i = 0; i < n - k; ++i) { nums.push_back(nums[0]); nums.erase(nums.begin()); } } };