1089-复写零

前言

Weekly Contest 141的 复写零：

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

解题思路

本题十分简单，需要注意本题是一个原地算法，所以使用双指针法进行变量交换即可。大致逻辑如下：

1. 遍历数组元素
2. 当前元素为0时，下一个元素赋值为0，并对数组进行左移，下一次遍历当前元素后2位的元素。否则正常遍历数组。

实现代码

    /**
     * 1089. 复写零
     * @param arr
     */
    public void duplicateZeros(int[] arr) {
        for (int i = 0; i < arr.length - 1;) {
            if (arr[i] == 0) {// 当前元素为0
                // 记录下一个元素的值
                int next = arr[i + 1];
                // 将下一个元素设为0
                arr[i + 1] = 0;
                for (int j = i + 2; j < arr.length; j++) { // 从当前索引后的第二个元素开始交换
                    int tmp = arr[j]; // 记录当前变量
                    arr[j] = next; // 将前一个位置的元素值赋给当前元素
                    next = tmp; // next指针记录当前元素
                }
                i+=2; // 下一次遍历的是当前元素后2位的元素
            }else{
                ++i; // 正常遍历
            }
        }
    }