リートコード1089。コピーゼロ

リートコード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を挿入します
• 最後に初期サイズの長さをインターセプトします

コード

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);
    }
};