给定一个数组,将数组中的元素向右移动 个位置,其中 是非负数。

示例 1:

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

输入: [-1,-100,3,99]k = 2
输出: [3,99,-1,-100]
解释: 
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

说明:

  • 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
  • 要求使用空间复杂度为 O(1) 的原地算法。

思路:

看了很多解法,要么就是时间复杂度太大,要么就是空间复杂度不为O(1)。

不过找到一个好解法,三次翻转数组即可。

例如:数组长度为len,移动数为k。

第一次翻转从0到len-k-1的部分;

第二次翻转从len-k到末尾的部分;

第三次翻转整个数组。

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