翻转数组(力扣)

题目:

给定一个整数数组 nums,将数组中的元素向右轮转 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]

示例 2:

输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释: 
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

思路:

想法一:

  • 向右轮转k个位置,就是将后k个数放到前面。
  • 先存储一个数组,将后k个数放置,再将剩余数组数依次向后挪动。
  • 再将k个数从前向后放置。
class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int len = nums.size();
        vector<int> newNums(k);
        if(len > k){
            for(int i = len - k,j = 0;i < len;++i,++j){
            newNums[j] = nums[i];
            }   
            for(int i = 0;i < len - k;++i){
                nums[len - 1 - i] = nums[len - k -1 -i];
            }
            for(int i = 0;i < k;++i){
                nums[i] = newNums[i];
            }
        }
        
        

    }
};
  • 第一次测试,没有考虑len < = k 的情况,产生了越界

  • 第二次测试,当k > len但确不是len的整数倍时,也会有变化。

    官方题解对于我这种方法的更好表述,提醒着我如何用数学更好的表达出代码,以后要减少这种多次for循环。

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

想法二:

开始看官方题解:

根据题意,如果使用多余数组存储空间,会导致空间复杂度为 nnn,所以在这里,我们可以使用常量级的空间复杂度解法:数组翻转。

思路如下:

  • 首先对整个数组实行翻转,这样子原数组中需要翻转的子数组,就会跑到数组最前面。
  • 这时候,从 kkk 处分隔数组,左右两数组,各自进行翻转即可。
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);
    }
};
作者:御三五 🥇

不得不说真的很奇妙,也解决了我上一个思路中k值大的情况,只要取余就可以解决。同时也解决了空间复杂度较大的问题。

另外,reverse函数可以直接调用,更方便。

posted @ 2023-02-24 17:04  ku然  阅读(33)  评论(0编辑  收藏  举报