leetcode 189.旋转数组(C++)

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

示例 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]


反转法

k可能比n大很多,所以需要对k %= n运算;

分组的时候,不是k的那个位置,而是要从后往前(n-k-1)

class Solution {
private:
    void reverse(vector& nums, int low, int high){
        while(low < high)
            swap(nums[low++], nums[high--]);
    }
public:
    void rotate(vector& nums, int k){
        long n = nums.size();
        k %= n;
        if (k == 0)
            return;
        reverse(nums, 0, n-k-1);
        reverse(nums, n-k, n-1);
        reverse(nums, 0, n-1);
    }
};

直接插入法

用vector的插入操作,将后面的数字插到前面去,需要resize

void rotate(vector& nums, int k) {
        if(k % nums.size() == 0)
            return;
        k %= nums.size();
        size_t size = nums.size();
        
        nums.insert(nums.begin(),nums.begin() + nums.size() - k,nums.end()); //insert(pos,begin,end)      
        nums.resize(size);
    }

 

你可能感兴趣的:(leetcode)