leetcode-數組-簡單-旋轉數組

題目

給定一個數組，將數組中的元素向右移動 k 個位置，其中 k 是非負數。

輸入: [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]

輸入: [-1,-100,3,99] 和 k = 2
輸出: [3,99,-1,-100]
解釋: 
向右旋轉 1 步: [99,-1,-100,3]
向右旋轉 2 步: [3,99,-1,-100]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/rotate-array
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

解答

方法一：

首先想到的就是窮舉操作每一個元素的的方法。

分析如下：

• 移動次數和數組長度相等，相當於沒有移動。因此對於K大於數組長度的情況需要進行求餘操作。
• 每次記錄最後一個元素，然後進行移動操作。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if (nums.empty() || nums.size() == 1 ) {
            return;
        }
        if (nums.size() < k) {
            k = k % nums.size();
        }
        for (int i = 0; i < k; i++) {
            int last_ele = nums.back();
            ///數據後退的一個操作，依次後退 
            for (int j = nums.size() - 1; j >= 1; --j) {
                nums[j] = nums[j - 1];
            }
            nums[0] = last_ele;
        }
    }
};

這個方案在leetcode提交之後是運行超時的。 

方法二：

因爲第一個辦法不行了，所以繼續想。

• 其實就是把數組分成兩個部分，然後把兩個部分進行交換，後邊的換到前邊。
• 後邊的怎麼換到前邊呢，翻轉一下就到前邊了，但是這樣兩個小部分的順序也是反的。
• 那針對兩個小部分各自做一次反轉就可以了。

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