LeetCode - 189. Rotate Array C++

最新推荐文章于 2020-10-03 16:33:57 发布

原创最新推荐文章于 2020-10-03 16:33:57 发布 · 242 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#array

LeetCode 专栏收录该内容

180 篇文章

订阅专栏

本文介绍了三种不同的算法来实现数组元素的旋转操作。第一种方法通过创建一个新的数组并使用模运算来放置元素，第二种方法则在不使用额外数组的情况下通过记录被覆盖的值进行元素的旋转，第三种方法利用反转技巧，先反转前半部分，再反转后半部分，最后反转整个数组，从而达到旋转的效果。

解法一 move to a new array

移动后的index可以用 (i+k)%len [i: current index; k: 移动的位数; len: size of the array]来计算出。

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

解法二 no extra array

基本思想是用一个variable来记录上一个被覆盖的值

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int len = nums.size();
        if (nums.empty() || (k %= len) == 0) return;       
        int idx=0, start=0, cur = 0, pre = nums[idx];
        for(int i=0;i<len;i++){
            int nidx = (idx+k)%len;
            cur = nums[nidx];
            nums[nidx] = pre;
            pre = cur;
            idx=nidx;
            if(idx==start){
                idx = ++start;
                pre = nums[idx];
            }
        }
    }
};

解法三 reverse

trick：reverse前半段，reverse后半段，reverse整个 ——> DONE

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int len = nums.size();
        if (nums.empty() || (k %= len) == 0) return;
        int n = (len-k)%len;
        reverse(nums.begin(), nums.begin()+n);
        reverse(nums.begin()+n, nums.end());
        reverse(nums.begin(), nums.end());
    }
};