日新计划Leetcode之 283. 移动零

89 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第10天,点击查看活动详情

一、题目描述:

283. 移动零 - 力扣(LeetCode)

给定一个数组 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种:

  1. 为0:则满足[li, index)区间的元素都是0,index++
  2. !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++;
    }
};

四、参考:

283. 移动零[题目说不能return 但是没说不可以替换] - 移动零 - 力扣(LeetCode)