LeetCode1089-复写零

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/duplicate-zeros/

中文链接：https://leetcode-cn.com/problems/duplicate-zeros/

题目详述

给你一个长度固定的整数数组 arr，请你将该数组中出现的每个零都复写一遍，并将其余的元素向右平移。

注意：请不要在超过该数组长度的位置写入元素。

要求：请对输入的数组 就地 进行上述修改，不要从函数返回任何东西。

示例 1：

1
2
3
输入：[1,0,2,3,0,4,5,0]
输出：null
解释：调用函数后，输入的数组将被修改为：[1,0,0,2,3,0,0,4]

示例 2：

1
2
3
输入：[1,2,3]
输出：null
解释：调用函数后，输入的数组将被修改为：[1,2,3]

提示：

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

题目详解

  • 定义快慢指针，扫描两次数组。
  • 第一次扫描定位复制结束的位置。首先初始化两个指针为 0，然后根据题意更新两个指针，直到快指针越过数组下标范围。
  • 重置快慢指针位置，从后往前扫描，开始复制。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
public class LeetCode_01089 {

    public void duplicateZeros(int[] arr) {
        int n = arr.length;
        int i = 0, j = 0;
        while (j < n) {
            if (arr[i] == 0) ++j;
            ++i;
            ++j;
        }
        --i;    // i 回到最后一次合法的位置
        --j;    // j 同理，但 j 仍可能等于 n（例如输入 [0]）
        while (i >= 0) {
            if (j < n) arr[j] = arr[i];
            if (arr[i] == 0) arr[--j] = arr[i];
            --i;
            --j;
        }
    }
}