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); } };
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理