189. 旋转数组
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
进阶:
- 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
- 你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
示例 1:
输入: 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 2 * 104-231 <= nums[i] <= 231 - 10 <= k <= 105
方法一:数组API
- array.splice(start[, deleteCount[, item1[, item2[, ...]]]]):通过删除或替换现有元素或者原地添加新的元素来修改数组,并以数组形式返回被修改的内容。此方法会改变原数组。
- unshift:将一个或多个元素添加到数组的开头,并返回该数组的新长度(该方法修改原有数组)
1 /** 2 * @param {number[]} nums 3 * @param {number} k 4 * @return {void} Do not return anything, modify nums in-place instead. 5 */ 6 var rotate = function(nums, k) { 7 k%=nums.length; 8 nums.unshift(...nums.splice(nums.length-k)); 9 };
方法二:使用额外的数组
思路:用 n表示数组的长度,我们遍历原数组,将原数组下标为 ii 的元素放至新数组下标为 n(i+k)modn 的位置,最后将新数组拷贝至原数组即可
1 /** 2 * @param {number[]} nums 3 * @param {number} k 4 * @return {void} Do not return anything, modify nums in-place instead. 5 */ 6 var rotate = function(nums, k) { 7 const n = nums.length; 8 const newArr = new Array(n); 9 for(let i = 0;i<n;i++){ 10 newArr[(i+k)%n] = nums[i]; 11 } 12 for(let j = 0;j<n;j++){ 13 nums[j] = newArr[j]; 14 } 15 };
方法三:环状替换
1 /** 2 * @param {number[]} nums 3 * @param {number} k 4 * @return {void} Do not return anything, modify nums in-place instead. 5 */ 6 const gcd = (x,y) => y ? gcd(y, x%y) : x; 7 var rotate = function(nums, k) { 8 const n = nums.length; 9 k = k%n; 10 let count = gcd(k,n); 11 for(let start = 0; start < count; ++start){ 12 let current = start; 13 let prev = nums[start]; 14 do{ 15 const next = (current + k) % n; 16 const temp = nums[next]; 17 nums[next] = prev; 18 prev = temp; 19 current = next; 20 }while(start !==current); 21 } 22 };
方法四:数组翻转
思路:我们可以先将所有元素翻转,这样尾部的kmodn 个元素就被移至数组头部,然后我们再翻转[0,kmodn−1] 区间的元素和[kmodn,n−1] 区间的元素即能得到最后的答案。
1 /** 2 * @param {number[]} nums 3 * @param {number} k 4 * @return {void} Do not return anything, modify nums in-place instead. 5 */ 6 var reverse = function(nums, start, end) { 7 while(start < end){ 8 const temp = nums[start]; 9 nums[start] = nums[end]; 10 nums[end] = temp; 11 start+=1; 12 end -=1; 13 } 14 }; 15 var rotate = function(nums,k){ 16 k%=nums.length; 17 reverse(nums, 0, nums.length-1); 18 reverse(nums, 0, k-1); 19 reverse(nums, k, nums.length-1); 20 }

浙公网安备 33010602011771号