📜  门| GATE-CS-2014-(Set-2) |第 50 题(1)

📅  最后修改于: 2023-12-03 15:12:42.182000             🧑  作者: Mango

题目描述

请实现一个函数,将一个整数数组中的每个元素向右移动 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]
解法
代码

以下是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 %= len(nums)
        nums[:] = nums[-k:] + nums[:-k]
解析
  • 首先将数组翻转。
  • 然后再将前k个元素翻转。
  • 最后再将后面的元素翻转。

Python中nums[:]表示对原列表进行操作而无需返回值。

复杂度分析
  • 时间复杂度:$O(n)$,其中$n$是数组的长度。
  • 空间复杂度:$O(1)$。
总结

本题给出了一种时间复杂度和空间复杂度均为$O(1)$的解法。同时这种解法与数组的长度无关,因此适用于在空间和时间方面均有限制的情况下。