老卫带你学---leetcode刷题(189. 轮转数组)

老卫带你学

于 2024-03-15 20:43:15 发布

阅读量409

点赞数 8

分类专栏： leetcode刷题文章标签： leetcode

本文链接：https://blog.csdn.net/yixieling4397/article/details/136749323

版权

leetcode刷题专栏收录该内容

120 篇文章 0 订阅

订阅专栏

189. 轮转数组

问题

给定一个整数数组 nums，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

示例 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]

解决

比较简单的思路就是pop + insert，但我觉得没意思

其实好玩点的玩法是：

先把所有的元素打个颠倒
再根据k值分成左数组和右数组
对左右数组分别再做一次颠倒

这样的空间复杂度（O（1））

    def rotate(self, nums, k):
        n = len(nums)
        k%=n
        for i in range(n//2):
            nums[i],nums[n-1-i]=nums[n-1-i],nums[i]
        for i in range(k//2):
            nums[i],nums[k-1-i]=nums[k-1-i],nums[i]
        for i in range(k,(n+k)//2):
            nums[i],nums[k+n-1-i]=nums[n+k-1-i],nums[i]