zhangdizhangdi

189.轮转数组

题目

LeetCode 中等

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

示例：
输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

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

题解

方法：新数组

O(n) O(n)

js
function rotate(nums, k) {
  let len = nums.length
  let newArr = new Array(len)
  for (let i = 0; i < len; i++) {
    newArr[(i + k) % len] = nums[i]
  }
  for (let i = 0; i < len; ++i) {
    nums[i] = newArr[i]
  }
}

const nums = [-1, -100, 3, 99],
  k = 2
rotate(nums, k)
console.log('🌰', nums)
执行结果
🌰 [ 3, 99, -1, -100 ]

方法：原地

O(n) O(1)

js
function rotate(nums, k) {
  const len = nums.length
  k %= len

  const reverse = (nums, start, end) => {
    while (start < end) {
      ;[nums[start], nums[end]] = [nums[end], nums[start]]
      start += 1
      end -= 1
    }
  }

  reverse(nums, 0, len - 1) //翻转所有元素
  console.log('翻转所有元素后：', nums)
  reverse(nums, 0, k - 1) //翻转 [0,k-1] 区间的元素
  console.log('翻转 [0,k-1] 区间：', nums)
  reverse(nums, k, len - 1) //翻转 [k,n-1] 区间的元素
  console.log('翻转 [k,n-1] 区间：', nums)
}

const nums = [1, 2, 3, 4, 5, 6, 7],
  k = 10
rotate(nums, k)
console.log('🌰', nums)
执行结果
翻转所有元素后： [
  7, 6, 5, 4,
  3, 2, 1
]
翻转 [0,k-1] 区间： [
  5, 6, 7, 4,
  3, 2, 1
]
翻转 [k,n-1] 区间： [
  5, 6, 7, 1,
  2, 3, 4
]
🌰 [
  5, 6, 7, 1,
  2, 3, 4
]