LeetCode算法个人解答——189.旋转数组

题目

给定一个数组，将数组中的元素向右移动 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（执行时间168ms，21.88%）

解答思路

1.通过示例，可以看出，每次旋转都是将最后的元素取出再插入到第0位
2.数组的 insert 操作可以指定元素的插入位置
3.数组的 pop 操作可以删去最右边（最后一位）元素
注：直接 return 的话，返回的是一个元组，而要求是得到数组。
数组是可变类型，内存地址指向数组本身，即内存地址不会因为修改数组内的元素而改变，所以不需要返回值。

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.
        """
        # 创建变量 i ，用来计算执行的次数
        i = 0
        # 执行 k 次
        for i in range(k):
            # 获得数组的最后一位的值
            value = nums[-1]
            # 删去数组的最后一个元素
            nums.pop()
            # 在第 0 位插入删去的元素的值
            nums.insert(0, value)
        # 返回 nums 的值
        # return nums

解2（执行时间80ms，参考别人写的）

解题思路

1.通过示例，可以发现整个过程分为两个部分
2.一个部分是从尾部移到头部的元素（要删去的元素）
3.一个部分是被移到头部的元素往后挤的元素（只需要移动位置的元素）
4.数组的切片可以将一个数组切成两个部分

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.
        """
        # 获取数组的长度
        num_len = len(nums)
        # 切片获得两组数组，再重新组合
        # nums[num_len - k:]获得最后k位的元素（要删去的元素）
        # nums[:num_len - k]获得其他元素（只需要移动位置的元素）
        nums[:] = nums[num_len - k:] + nums[:num_len - k]

解3（解2升级版，执行时间64ms）

解题思路

1.考虑到移动次数可能会多余数组本身的长度，消去多余的部分，只留下实际移动的次数
2.其他和解2相同

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.
        """
        # 获得元素实际移动的次数
        k = k % len(nums)
        # 切片获得两组数组，再重新组合
        # nums[- k:]获得最后k位的元素（要删去的元素）
        # nums[:- k]获得其他元素（只需要移动位置的元素）
        nums[:] = nums[-k:] + nums[:-k]

解4（最佳解，想法新颖，执行时间52ms）

解题思路

1.考虑到移动次数可能会多余数组本身的长度，消去多余的部分，只留下实际移动的次数
2.取得只需要移动位置的元素，并加入到数组的尾部
3.删去前面只需要移动位置的元素的部分

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.
        """
        # 获去数组 nums 的长度
        l = len(nums)
        # 判断移动次数是否没动或移动的次数等于数组本身长度
        if k == 0 or k == l:
            # 若符合条件,不需要执行
            return
        k = k % l
        # 将只需要移动位置的元素添加到数组 nums 的尾部
        nums.extend(nums[:l-k])
        # 删除前l-k位的元素(只需要移动位置的元素)
        del nums[:l-k]