189. 轮转数组
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 1:
输入: nums = [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]
=====================================================
看到这个题目我直观的想要用另一个数组来保存要轮转的之后的元素,之后将轮转之前的元素移到后面,再将轮转之前的元素赋值到数组,上代码:
class Solution { public: void rotate(vector<int>& nums, int k) { while (nums.size() < k) { k = k - nums.size(); } int t = nums.size() - k ; vector<int> n; for (int i = t; i < nums.size(); i++) { n.push_back(nums[i]); } for (int i = t - 1; i >= 0; i--) { nums[i + k] = nums[i]; } for (int i = 0; i < k; i++) { nums[i] = n[i]; } } };
这个方法时间复杂度是O(1),空间复杂度是O(n);
当我看到官方的解法中的数组翻转,被惊艳到了
学到了,上代码:
class Solution { public: void reverse(vector<int>& nums, int start, int end) { while (start < end) { swap(nums[start], nums[end]); start += 1; end -= 1; } } void rotate(vector<int>& nums, int k) { k %= nums.size(); reverse(nums, 0, nums.size() - 1); reverse(nums, 0, k - 1); reverse(nums, k, nums.size() - 1); } };