1. 程式人生 > 其它 >python對齊輸出

python對齊輸出

技術標籤:資料結構與演算法leetcode資料結構演算法

給定一個數組,將陣列中的元素向右移動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

輸出:[3,99,-1,-100]
解釋:
向右旋轉 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