题目
给你一个数组,将数组中的元素向右轮转 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
网友评论