持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第10天,点击查看活动详情
一、题目描述:
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
提示:
- 1 <= nums.length <= 10^4
- -2^31 <= nums[i] <= 2^31 - 1
进阶:你能尽量减少完成的操作次数吗?
二、思路分析:
这道题思路上有点像三向切分的快速排序,将待排序元素基于基准值划分为小于,等于,大于三组。对于一次排序扫描需要3个指针:
- 低位指针li:< li对应的元素都小于基准元素
- 高位指针hi:> hi对应的元素都大于基准元素
- 当前遍历元素的指针index 并且[li, hi](闭区间)的元素都是相等的
而这题只有2种情况,0和!0(相当于小于,等于),所以只需要两个指针:
- 低位指针li:< li对应的元素都是!0的
- 当前遍历元素的指针index
在index往右遍历过程中,[li, index)区间的元素都是0,>= index的都是未确定元素,判断index元素的情况也只有2种:
- 为0:则满足[li, index)区间的元素都是0,index++
- !0:则与li交互位置,同时li++后,满足< li对应的元素都是!0的,也满足上面一点。
三、AC 代码:
class Solution {
function moveZeroes(nums: number[]): void {
const len = nums.length;
let li = null; // 放在左边起第一个0所在的位置,小于该指针的序号对应的元素都是`!0`的
let index = 0; // 当前遍历的元素
while (index < len) {
// 这个判断只是用来找第一个0的位置,不用关注...
if (nums[index] === 0 && li == null) {
li = index;
}
// 把右侧!0元素都与左边的0所在位置交换,相当于上面的第2点
else if (nums[index] && li != null) {
nums[li] = nums[index];
nums[index] = 0;
li++;
}
index++;
}
};