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 <= arr.length <= 10000
0 <= arr[i] <= 9

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/duplicate-zeros
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

感觉该题难度够个medium-了。先找到复写后保留在新数组中最后一个的数，然后将这个数之前的数倒着写入数组。具体细节见注释。

代码

双指针

class Solution {
    public void duplicateZeros(int[] arr) {
        int n = arr.length;
        // 计数，累加够n个就停止
        int cnt = 0;
        // 复写0后数组可以存放的最后一个数（也就是说，原数组k下标之后的数就被抛弃了）
        int k = -1;
        while (cnt < n) {
            k++;
            if (arr[k] == 0) {
                cnt += 2;
            } else {
                cnt++;
            }
        }
        // 接下来，倒着遍历arr[0,k]，并倒着赋值到arr[0,n-1]
        int i = n - 1;
        // cnt在这里可能为n，也可能为n+1
        // 如果为n+1，那么说明arr[k]=0，但是这个0只能写入一个，因此不能使用下面的逻辑，这里单独将arr[n-1]赋值为0
        if (cnt == n + 1) {
            k--;
            i--;
            arr[n - 1] = 0;
        }
        while (k >= 0) {
            arr[i--] = arr[k];
            if (arr[k] == 0) {
                arr[i--] = 0;
            }
            k--;
        }
    }
}