LeetCode:旋转数组
最近看了一道题,自己做个过后又参考了网上的解法,为了加深对这个解法的理解和记忆于是有了这篇博客,供自己以后复习用
题目:
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 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) 的 原地 算法。
解法:
1 var rotate = function(nums, k) { 2 var temp1, temp2, index, count = 0; // temp1存要换到后边的值,temp2存后边的值 3 var len = nums.length 4 k %= len 5 if (k !== 0 ) { 6 for (var i = 0; i <= k; i++) { // 为什么最多k轮,因为照这种方法一个一个挪最多就是k轮,再挪就是挪挪过的元素了 7 if (count >= len) break; 8 index = i; 9 temp1 = nums[i]; 10 while((index + k) % len !== i) { // i在while出来之前一直没变,所以当 (index + k) % len === i 时,把从里面出来的temp1给nums[i],如17行 11 temp2 = nums[(index + k) % len] 12 nums[(index + k) % len] = temp1 13 index = (index + k) % len 14 temp1 = temp2 15 count++ 16 } // 只有当数组中要换到后面的那个元素的下标===开始循环是的下标的时候跳出循环 17 nums[i] = temp1 18 count++ 19 } 20 console.log(nums) 21 } 22 }; 23 rotate([1, 2, 3, 4, 5, 6, 7], 3)