python對齊輸出
阿新 • • 發佈:2021-02-05
給定一個數組,將陣列中的元素向右移動k個位置,其中k是非負數。
進階:
儘可能想出更多的解決方案,至少有三種不同的方法可以解決這個問題。
你可以使用空間複雜度為O(1) 的原地演算法解決這個問題嗎?
示例 1:
輸入: nums = [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:
輸入:nums = [-1,-100,3,99], k = 2
解釋:
向右旋轉 1 步: [99,-1,-100,3]
向右旋轉 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
解題思路:
陣列翻轉。
可以先將所有元素翻轉,這樣尾部的k mod n 個元素就被移至陣列頭部,然後我們再翻轉 [0,k mod n−1] 區間的元素和 [k mod n,n−1] 區間的元素即能得到最後的答案。
Python程式碼:
class Solution: def rotate(self, nums: List[int], k: int) -> None: """ Do not return anything, modify nums in-place instead. """ k = k % len(nums) self.reverse(nums, 0, len(nums)-1) self.reverse(nums, 0, k-1) self.reverse(nums, k, len(nums)-1) def reverse(self, nums, l, r): while l < r: nums[l], nums[r] = nums[r], nums[l] l += 1 r -= 1