189. 旋转数组

最新推荐文章于 2021-12-20 19:18:01 发布

原创最新推荐文章于 2021-12-20 19:18:01 发布 · 117 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode

刷题记录专栏收录该内容

79 篇文章

订阅专栏

这篇博客探讨了如何在LeetCode的‘旋转数组’问题中，使用两种不同方法实现原地旋转数组。第一种方法通过三次翻转实现空间复杂度为O(1)的解决方案，第二种方法则利用新数组达到旋转目的，但空间复杂度为O(n)。这两种方法都详细解释了其工作原理和代码实现。

题目描述：

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。

进阶：

尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的原地算法解决这个问题吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotate-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题分析：

空间复杂度为O(1)：
1.翻转所有元素
2.翻转 [0, k mod n−1] 区间的元素
3.翻转 [k mod n−1，nums.size()-1] 区间的元素

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if(nums.size()<=1)
            return ;
        if(k>nums.size())
            k=k%nums.size();
        reverse(nums, 0, nums.size()-1);
        reverse(nums, 0, k-1);
        reverse(nums, k, nums.size()-1);
        return ;
    }

    void reverse(vector<int>& nums, int start, int end){
        while(start<end){
            int temp=nums[start];
            nums[start]=nums[end];
            nums[end]=temp;
            start++;
            end--;
        }
        return ;
    }
};

空间复杂度为O(n)：
将原数组下标为 ii 的元素放至新数组下标为 (i+k) mod n 的位置

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> newArr(n);
        for (int i = 0; i < n; ++i) {
            newArr[(i + k) % n] = nums[i];
        }
        nums.assign(newArr.begin(), newArr.end());
    }
};