[LeetCode][C++]旋转数组
程序员文章站
2022-04-28 09:54:54
...
旋转数组
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
思路:
- 旋转次数k,应当对数组大小n求余,以减少重复的旋转。即k=k%n,后续步骤的k均为求余后的k。
- 首先将整个数组反转,将后k个元素转到前k个。
- 对前k个元素再进行反转,以保持原数组中元素的相对位置。
- 对后n-k个元素再进行反转,以保持原数组中元素的相对位置。
代码:
class Solution {
public:
//反转函数
void reverse(vector<int>& nums,int start,int end)
{
int tmp;
while(start<end)
{
tmp=nums[start];
nums[start++]=nums[end];
nums[end--]=tmp;
}
}
void rotate(vector<int>& nums, int k) {
//预处理k,很重要。
k=k%nums.size();
//按步骤反转
reverse(nums,0,nums.size()-1); //反转整个数组
reverse(nums,0,k-1); //反转前k个
reverse(nums,k,nums.size()-1); //反转后面部分
}
};
结果:
参考链接:
[1] 潮汐:旋转数组(rotate-array) – 4种方法
上一篇: win10 uwp 商业游戏 1.2.1
下一篇: 当下站长群体的归宿地