LeetCode 189. 轮转数组

题目

给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。

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

方法

思路同 剑指 Offer 58. II. 左旋转字符串,区别在于本题向右移动,那么可以通过翻转前 n-k 个字符,翻转后 k 个字符,翻转整个字符串来实现

class Solution(object):
    def rotate(self, nums, k):
        def reverse(i, j):
            while i < j:
                nums[i], nums[j] = nums[j], nums[i]
                i += 1
                j -= 1
        n = len(nums)
        k %= n    
        reverse(0, n - k - 1)
        reverse(n - k, n - 1)
        reverse(0, n - 1)
参考

代码相关:https://programmercarl.com/0189.%E6%97%8B%E8%BD%AC%E6%95%B0%E7%BB%84.html

你可能感兴趣的:(LeetCode 189. 轮转数组)