【力扣每日一题】1089. 复写零

题目描述

给你一个长度固定的整数数组 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. 1 <= arr.length <= 10000
2. 0 <= arr[i] <= 9

---

复写0，相当于插入一个0，于是我联想到数据结构中 线性表的插入操作，先将插入位置后面的元素整体往后移一位，然后插入目标元素。要注意的是，插入后，索引需要向后移动一位，不然可就全都变成0了。

class Solution {
    public void duplicateZeros(int[] arr) {
        int n = arr.length;
        for(int i=0;i<n-1;i++){
            if(arr[i]==0){
                insert(arr,i);
                i++;
                
            }
        }
    }
    public void insert(int[] arr,int idx){
        int n = arr.length;
        for(int i=n-1;i>idx;i--){
            arr[i] = arr[i-1];
        }
    }
}

---

官方演示了用栈如何保存答案，然后反向输入到数组中，这个思路很好，进而又推理出可以用双指针替代栈进行top标记，然后返回答案，妙！

class Solution {
    public void duplicateZeros(int[] arr) {
        int n = arr.length;
        int top = 0;
        int i = -1;
        //算出i（索引）的值，即最后的答案到哪个位置停止
        while (top < n) {
            i++;
            if (arr[i] != 0) {
                top++;
            } else {
                top += 2;
            }
        }
        //最后一个为0（超范围）时的处理方式
        int j = n - 1;
        if (top == n + 1) {
            arr[j] = 0;
            j--;
            i--;
        } 
        //重新填充arr
        while (j >= 0) {
            arr[j] = arr[i];
            j--;
            //等于0时填充2次
            if (arr[i] == 0) {
                arr[j] = arr[i];
                j--;
            } 
            i--;
        }
    }
}