LeetCode 189.Rotate Array (旋轉陣列)
阿新 • • 發佈:2018-11-20
題目描述:
給定一個數組,將陣列中的元素向右移動 k 個位置,其中 k 是非負數。
示例 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]
示例 2:
輸入: [-1,-100,3,99] 和 k = 2 輸出: [3,99,-1,-100] 解釋: 向右旋轉 1 步: [99,-1,-100,3] 向右旋轉 2 步: [3,99,-1,-100]
說明:
- 儘可能想出更多的解決方案,至少有三種不同的方法可以解決這個問題。
- 要求使用空間複雜度為 O(1) 的原地演算法。
解題思路:
** 輸入: [1,2,3,4,5,6,7] 和 k = 3
** 第一次翻轉: 4 3 2 1 5 6 7
** 第二次翻轉: 4 3 2 1 7 6 5
** 整體翻轉: 5 6 7 1 2 3 4 輸出
AC C++ Solution:
class Solution { public: void rotate(vector<int>& nums, int k) { int n=nums.size(); if(nums.empty()||(k=k%n)==0) return ; //翻轉三次 reverse(nums.begin(),nums.begin()+n-k); //反轉前n-k個元素 reverse(nums.begin()+n-k,nums.end()); //反轉後k個元素 reverse(nums.begin(),nums.end()); //整體反轉 } };