leetcode系列-旋转数组(看到这个题解,我都惊呆了系列)

分类:array

难度:easy

给定一个数组,将数组中的元素向右移动 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(反转法)

我们可以看到往右移动k个单位,原本后k个元素调到开头,前面n-k个元素依次往后推,所以可以将数组全部反转,然后将前k个反转,再将剩下的n-k个元素反转即可
时间复杂度O(n)
空间复杂度O(1)

代码

class Solution(object):
    def rotate(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        # 反转法
        k = k%len(nums)
        def reverse(nums,left,right):
            curl,curr = left,right
            while curl<curr:
                nums[curl],nums[curr]=nums[curr],nums[curl]
                curl+=1
                curr-=1
        reverse(nums,0,len(nums)-1)
        reverse(nums,0,k-1)
        reverse(nums,k,len(nums)-1)

题解2(环状替换)

晚上补充一下

发布了41 篇原创文章 · 获赞 0 · 访问量 6159

猜你喜欢

转载自blog.csdn.net/Yolo_C/article/details/104772112