欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

[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) 的 原地 算法。


思路:

  1. 旋转次数k,应当对数组大小n求余,以减少重复的旋转。即k=k%n,后续步骤的k均为求余后的k。
  2. 首先将整个数组反转,将后k个元素转到前k个。
  3. 对前k个元素再进行反转,以保持原数组中元素的相对位置。
  4. 对后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);     //反转后面部分       
    }
};

结果:
[LeetCode][C++]旋转数组

参考链接:
[1] 潮汐:旋转数组(rotate-array) – 4种方法

相关标签: LeetCode