算法笔记 -- 1089. 复写零

71 阅读2分钟

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

一、题目描述:

1089. 复写零 - 力扣(LeetCode)

给你一个长度固定的整数数组 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]

提示:

  • 1 <= arr.length <= 10000
  • 0 <= arr[i] <= 9

二、思路分析:

首先,可以用模拟解决这个问题,但是这样很显然需要o(n) 的空间复杂度
基于这个思路想一下,我们是否一定要开一个数组存结果呢?
我们知道,如果有零的话,末尾的一些元素会被抛弃掉,所以我们可以直接往那些被抛弃的位置填结果
容易想到,首先扫一下数组,计算一下填充后的队尾是位置ptr
然后搞一个tail指针指向队尾最后的位置,一直将 arr[ptr] 写入到arr[tail],如果遇到0,就额外写一个零

最后需要注意的是,如果末尾是个0,它可能不会复写

三、AC 代码:

class Solution {
    public void duplicateZeros(int[] arr) {
        int[] next = new int[arr.length];

        int ptr = 0, i = 0;
        while(ptr < arr.length) {
            next[ptr++] = arr[i];
            if (arr[i] == 0 && ptr < arr.length) {
                next[ptr++] = arr[i];
            }
            ++i;
        }

        for (i = 0; i < arr.length; ++i) {
            arr[i] = next[i];
        }
    }
}

三、参考:

使用map记录非零元素之前的0的个数,arr填充0,遍历map重新赋值 - 复写零 - 力扣(LeetCode)

我的C++ 转成string去操作有点麻烦,有没有大佬帮忙优化一下? - 复写零 - 力扣(LeetCode)