Leetcode 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]

思路

• 遇到一个0就在当前的位置插入一个0
• 最后截取初始size长度

代码

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        int size = arr.size();

        for (int i = 0;i < size;++i) {
            if (arr[i] == 0) {
                arr.insert(arr.begin() + i, 0);
                ++i;
            }
        }

        arr = vector<int>(arr.begin(), arr.begin() + size);
    }
};