【leetcode】189. 旋轉數組( Rotate Array )

---

題目描述

【leetcode】189. 旋轉數組( Rotate Array )

給定一個數組，將數組中的元素向右移動 k 個位置，其中 k 是非負數。
說明:
儘可能想出更多的解決方案，至少有三種不同的方法可以解決這個問題。
要求使用空間複雜度爲 O(1) 的 原地 算法。

第一次解答

思路：
題意是把數組nums向右平移k次，超出邊界的部分從數組首插入。
簡單的方法是按部就班的平移1次，2次，…，一直到k次，該方法缺點就是每次平移所有元素都要移動一次，複雜度是k*nums.size()。
另一種思路是一步計算得到每個元素最終所在的位置，將元素直接放到最終位置。用一個臨時變量保存原位置值，但是這個代碼不好實現，我寫了幾次都有bug，最後還是參考了官方題解。

test case:
[]
1
[1]
1
[1,2]
1
[1,2]
2
[1,2]
3
[1,2,3,4,5,6,7]
3
[1,2,3,4,5,6]
2

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if(nums.size() == 0)
            return;
        k = k%nums.size(); // 若k>=nums.size()，平移k次與平移k%nums.size()次效果相同
        if(0 == k)
            return;
        
        int count = 0; // 記錄已變換位置的元素個數
        // 注意這個for循環執行的判斷條件
        // 若 nums.size() % k == 0， 則for循環執行k次，直到count==nums.size()
        // 若 nums.size() % k != 0， 則for循環執行1次（因爲在do-while裏一直執行到count==nums.size()）。
        for(int start=0; count<nums.size(); ++start){
            int current = start;
            int last_value = nums[current];
            int next;
            // 注意這裏用do-while而不是wihle
            do{
                next = (current + k) % nums.size();
                int temp = nums[next];
                nums[next] = last_value;
                last_value = temp;
                current = next;
                ++count;
            }while(start != current);
        }
    }
};

結果：

截圖

相關/參考鏈接