leetcode189 轮转数组

60 阅读1分钟

给定一个整数数组 nums,将数组中的元素向右轮转 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]

数组翻转

输入 nums = "----->-->"; k =3

最后结果为 "-->----->";

翻转 "----->-->" 得到 "<--<-----"

翻转 "<--" 得到 "--><-----"

翻转 "<-----" 得到 "-->----->"

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var rotate = function(nums, k) {
    k %= nums.length;
    // nums整体翻转
    reverse(nums, 0, nums.length - 1);
    // 翻转前k个
    reverse(nums, 0, k - 1);
    // 翻转剩下的
    reverse(nums, k, nums.length - 1);
};

// 翻转数组的函数
const reverse = (nums, start, end) => {
    while (start < end) {
        const temp = nums[start];
        nums[start] = nums[end];
        nums[end] = temp;
        start += 1;
        end -= 1;
    }
}

splice 后 unshift() 塞入

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var rotate = function (nums, k) {
  if (k > nums.length) {
    k = k % nums.length
  }
  let arr = nums.splice(nums.length - k , k )
  for (let i = arr.length-1 ; i >= 0 ; i--) {
    nums.unshift(arr[i])
  }
};

slice 保存 再放入

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var rotate = function (nums, k) {
  if (k > nums.length) {
    k = k % nums.length
  }
  let arr = nums.slice(nums.length - k, nums.length)
  for (let i = nums.length - 1; i > k - 1; i--) {
    nums[i] = nums[i - k]
  }
  for (let j = 0; j < arr.length; j++) {
    nums[j] = arr[j]
  }
};