第五次實驗
阿新 • • 發佈:2021-10-11
給定一個數組,將陣列中的元素向右移動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]
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/rotate-array
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
class Solution(object): def reverse(self,nums,i,j): while i<j: nums[i],nums[j]=nums[j],nums[i] i+=1 j-=1 def rotate(self, nums, k): """ :type nums: List[int] :type k: int :rtype: None Do not return anything, modify nums in-place instead.""" length=len(nums) k%=length ##會出現[-1] k=2的情況 self.reverse(nums,0,length-1) self.reverse(nums,0,k-1) self.reverse(nums,k,length-1)
難點在於如何寫reverse函式——其實只要雙指標頭尾互換就行了
傳入引數是雙指標分別開始的地方
還有一難點是特殊情況,已經標在註釋裡