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 - 1
  • 0 <= 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 }
 
posted @ 2021-08-09 19:26  icyyyy  阅读(91)  评论(0)    收藏  举报