力扣每日一题0617-1089. 复写零

80 阅读2分钟

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

给你一个长度固定的整数数组 arr,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。

注意:请不要在超过该数组长度的位置写入元素。

要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。

示例 1:

输入:[1,0,2,3,0,4,5,0]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,0,0,2,3,0,0,4]

示例 2:

输入:[1,2,3]
输出:null
解释:调用函数后,输入的数组将被修改为:[1,2,3]

我们可以假设已经通过原数组 arr 处理出最终的目标数组 ans,起始使用指针 ii 指向原数组的开头位置,使用指针 jj 指向目标数组的开头位置。

然后开始同时从前往后处理,默认情况下,两个指针每次往后移动一位,当在原数组中出现 arr[i]=0arr[i] = 0 时,根据题意,我们需要在目标数组 ans[j]ans[j] 后面复写多一位,因此此时让 jj 多走一位。

jj 走到的位置已超出数组长度,此时 ii 也停在了该被截断的位置的下一位。

此时我们先将 iijj 同时往回走一步,确保 ii 落在真正的截断后的位置,但需要注意,由于 jj 在之前的往前走逻辑中可能会走两步,因此 jj 在往回走一步后仍可能在数组以外的位置。

然后考虑如何在 iijj 所在位置开始「往回」赋值:每次将 arr[i]arr[i] 赋值给 arr[j]arr[j](需要确保 jj 下标合法),由于 ii 必然不会出现在 jj 的后面(即不会出现 j<ij<i ),因此不会出现值被提前覆盖的问题。因此通常情况下,我们可以不断的同时减少 iijj,并将 arr[i]arr[i] 赋值给 arr[j]arr[j]。当出现 arr[i]=0arr[i]=0 时,我们需要额外让 jj 往前多走一步,并将该位置置为 00,来实现目标数组「复写零」的效果。

ii 指针越过开头的位置,说明已完成从 arrans 的转换。

var duplicateZeros = function(arr) {
    const n = arr.length;
    let top = 0;
    let i = -1;
    while (top < n) {
        i++;
        if (arr[i] !== 0) {
            top++;
        } else {
            top += 2;
        }
    }
    let j = n - 1;
    if (top === n + 1) {
        arr[j] = 0;
        j--;
        i--;
    } 
    while (j >= 0) {
        arr[j] = arr[i];
        j--;
        if (arr[i] === 0) {
            arr[j] = arr[i];
            j--;
        } 
        i--;
    }
};