美文网首页
leetcode-数组-旋转数组|Rotate Array(Py

leetcode-数组-旋转数组|Rotate Array(Py

作者: lqy007700 | 来源:发表于2018-07-25 15:06 被阅读0次

    给定一个数组,将数组中的元素向右移动 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) 的原地算法。
    

    代码主体:

    两种解法大同小异,都是在评论区找到的(自愧不如)
    class Solution:
        def rotate(self, nums, k):
            """
            :type nums: List[int]
            :type k: int
            :rtype: void Do not return anything, modify nums in-place instead.
            """
    
            l = len(nums)
            if k % l != 0:
                nums[:k], nums[k:] = nums[l - k:], nums[:l - k]
    
        def rotate1(self, nums, k):
            while k > 0:
                nums.insert(0, nums.pop())
                k -= 1
    

    相关文章

      网友评论

          本文标题:leetcode-数组-旋转数组|Rotate Array(Py

          本文链接:https://www.haomeiwen.com/subject/uzbpmftx.html