[LeetCode #1089] 复写零

目录

题目描述 

解题思路

解题代码

---

题目描述 

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

注意：请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改，不要从函数返回任何东西。

示例 1：

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

示例 2：

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

提示：

• 1 <= arr.length <= 104
• 0 <= arr[i] <= 9

---

解题思路

从后向前复写,预留出需要复写0的位置即可

---

解题代码

C++:

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int cur = 0, dest = -1, size = arr.size();
        // 先预留出复写0的位置，当dest为n-1的时候跳出
        while(cur < size) {
            if (arr[cur]) { dest++; }
            else { dest += 2; }
            if (dest >= size - 1) { break; }
            ++cur;
        }
        // 2. 处理边界情况
        if (dest == size) {
            arr[size-1] = 0;
            --cur;
            dest -= 2;
        }
        // 3. 开始复写
        while (cur >= 0) {
            if (arr[cur]) { arr[dest--] = arr[cur--]; }
            else {
                arr[dest--] = 0;
                arr[dest--] = 0;
                --cur;
            }
        }
    }
};

Java:

class Solution {
    public void duplicateZeros(int[] arr) {
        int cur = 0, dest = -1, size = arr.length;
        // 先预留出复写0的位置，当dest为n-1的时候跳出
        while(cur < size) {
            if (arr[cur] != 0) { dest += 1; }
            else { dest += 2; }
            if (dest >= size - 1) { break; }
            ++cur;
        }
        // 2. 处理边界情况
        if (dest == size) {
            arr[size-1] = 0;
            cur--;
            dest -= 2;
        }
        // 3. 开始复写
        while (cur >= 0) {
            if (arr[cur] != 0) { arr[dest--] = arr[cur--]; }
            else {
                arr[dest--] = 0;
                arr[dest--] = 0;
                cur--;
            }
        }
    }
}