[LeetCode刷题]第189题 旋转数组(简单易懂,转置数组实现)

300 阅读2分钟

一、读题

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

示例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]

二、解题思路

首先我们假设需要旋转的数组的内容为nums= "----->-->" (为看起来直观,我们用字符串来表示这个数组),数组的长度为9 需要移动的位置为k = 12

那么旋转12次我们需要的结果就是 -->----->

1.计算需要旋转的次数

我们对k值也就是旋转的次数取余于数组的长度,那么便是我们需要旋转的次数

比如这次需要移动的位置是12,而长度为9

那么 k % nums.length 便是实际旋转的次数

2.根据移动位置转置数组

2.1转置整体

我们旋转整体

"----->-->"

将得到

"<--<-----"

2.2转置前半部分

我们旋转从0到k-1的部分 假设k为3,那么我们将取下标0到下标2的部分进行转置

从"<--<-----"取出 "<--"翻转

将得到

"--><-----"

2.3转置后半部分

我们旋转从k到数组最后一位的部分 假设k为3,那么我们将取下标3到下标的部分进行转置

从"<--<-----"取出 "<-----"翻转

将得到

"-->----->"

至此旋转完成,此结果便是我们所需要的结果

2.4换个方式呈现

nums = "----->-->"; k =3
result = "-->----->";

reverse "----->-->" we can get "<--<-----"
reverse "<--" we can get "--><-----"
reverse "<-----" we can get "-->----->"

三、实现方式

3.1 实现一个reverse

直接贴上代码 一篇文章: js双指针翻转数组指定部分 不使用内置函数

/*
* 传入nums为需要转置的数组,l1为起始坐标,l2为结束坐标
*/
const reverse = (nums,l1,l2) => {
	l1 = l1 || 0;
	l2 = l2 || nums.length-1;
	// 双指针遍历
    while(l1<l2) {
    	// 进行前后位置对调
        let temp = nums[l1];
        nums[l1] = nums[l2];
        nums[l2] = temp;
		// 更新下一组
        l1++;
        l2--;
    }
}

3.2 进行转置

const rotate = (nums, k) => {
    k = k % nums.length;
    reverse(nums,0,nums.length-1);
    reverse(nums,0,k-1);
    reverse(nums,k,nums.length-1);
    return nums;
};

四、完整代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {void} Do not return anything, modify nums in-place instead.
 */
const reverse = (nums,l1,l2)=> {
    while(l1<l2) {
        let temp = nums[l1];
        nums[l1] = nums[l2];
        nums[l2] = temp;

        l1++;
        l2--;
    }
}
const rotate = (nums, k) =>{
    k = k % nums.length;
    reverse(nums,0,nums.length-1);
    reverse(nums,0,k-1);
    reverse(nums,k,nums.length-1);
};

五、相关链接

1.LeetCode第189题

2.解题思路来源

3.感谢分享