LeetCode 探索初级算法-数组：03 旋转数组-20200316

03 旋转数组-20200316

题目

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。

示例

输入: [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]

说明

• 尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
• 要求使用空间复杂度为 O(1) 的 原地 算法。

---

注意事项

1. 循环移动，最笨的办法就是逐一移动，因为要求空间复杂度为O(1)，所以不能另外开辟新数组。
2. 题目不要求返回值，直接对数组操作。
3. 不用考虑数组长度和k的大小关系，就是移动就行。
4. 写出来三种算法。

思路一

先来个笨办法，在数组后面加一项，逐个向后移，再把最后一项移到最前面，这个方法的时间复杂度应该很高。

修改经历：

1. 我知道时间会很长，但是没想到超出了时间限制？！！头大，两层循环看来不行啊。（第一次提交）

心得体会：

这个方法，我没有在题解关于Python 3上看到，可能就是太笨了吧。

最终代码展示：

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        for i in range(k%len(nums)):
            temp = nums[-1]
            for j in range(len(nums)-1, 0, -1):
                nums[j] = nums[j-1]
            nums[0] = temp

思路二

我想我可能是个傻子。。。直接把数组后面的截取放到前面不好吗？但是这样就可能要新开辟一个数组了。

修改经历：

1. 没有考虑到输入数组为[1]，k=0的情况（第一次提交）

2.没有考虑到k大于数组长度的情况，比如[1]，k=2。（第二次提交）

• 执行用时 :36 ms, 在所有 Python3 提交中击败了95.39%的用户
• 内存消耗 :13.8 MB, 在所有 Python3 提交中击败了97.58%的用户

心得体会：

我是真的没有想到，思路二竟然能通过，而且排名还这么靠前。。。O(1)的空间复杂度呢？这不就是相当于O(N)的空间复杂度了吗？

看到一种非常简单的写法：

lenth = len(nums)
nums[:] = nums[lenth-k:]+nums[:lenth-k]

这个真的是牛逼了，连临时变量都不需要了

最终代码展示：

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        if k == 0 or len(nums) == 1:
            pass
        else:
            nums1 = nums[-k%(len(nums)):]
            nums2 = nums[:-k%(len(nums))]
            nums[:k%(len(nums))] = nums1
            nums[k%(len(nums)):] = nums2

思路三

这个思路是在题解上看到的，把最后一个pop()，再插入第一项。。。代码就两行。看来Python还是没玩转啊。

最终代码展示：

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)
        for i in range(k):
            nums.insert(0,nums.pop())

思路四

这是看题解的大神写得，我觉得这种是最复合题目要求的空间复杂度为O(1)的要求了。这就是翻转法，我把解法贴出来了。

三次反转
对于[1,2,3,4,5,6,7]，
根据k=k\%n，将数组分为两段：

• 第一段，对应数组下标范围[0,n-k-1]段，即[1,2,3,4]
• 第二段，对应数组下标范围[n-k,n-1]，即[5,6,7]

分为三步：

• 反转第一段，[4,3,2,1,5,6,7]
• 反转第二段，[4,3,2,1,7,6,5]
• 反转整体，[5,6,7,1,2,3,4]

完成！

心得体会：
这个方法妙哉啊，要好好记下来！这个是怎么想到的呢？

最终代码展示：

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        def swap(former, latter):
            while former < latter:
                nums[former], nums[latter] = nums[latter], nums[former]
                former += 1
                latter -= 1
        k = k % (len(nums))
        swap(0, len(nums)-k-1)
        swap(len(nums)-k, len(nums)-1)
        swap(0, len(nums)-1)