乐趣区

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; // 正常遍历}
        }
    }
退出移动版