提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  其他

当前栏目

Leetcode No.189 旋转数组

数组
2023-03-20 14:56:47 时间

一、题目描述

给定一个数组，将数组中的元素向右移动 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]

提示：

1 <= nums.length <= 2 * 10^4 -2^31 <= nums[i] <= 2^31 - 1 0 <= k <= 10^5

二、解题思路

我们可以使用额外的数组来将每个元素放至正确的位置。用 n 表示数组的长度，我们遍历原数组，将原数组下标为 i 的元素放至新数组下标为 (i+k)%n 的位置，最后将新数组拷贝至原数组即可。

三、代码

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.
        """
        n=len(nums)
        k=k%n
        t=[]
        j=0
        for i in range(n-k,n):
            t.insert(j,nums[i])
            j=j+1

        for i in range(k-1,n)[::-1]:
            nums[i]=nums[i-k]

        for i in range(0,k):
            nums[i]=t[i]

if __name__ == '__main__':
    array=[1,2,3,4]
    solution=Solution()
    n=len(array)
    solution.rotate(array,2)
    for i in range(0,n):
        print(array[i])

四、复杂度分析

时间复杂度： O(n)，其中 n 为数组的长度。

空间复杂度： O(n)。

相关文章