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]
 
提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/duplicate-zeros
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：重新开辟一个新空间 res，然后把每个0复制一份，截取res的长度与原arr长度一致，复制回去，即可。时间复杂度O(n)，空间复杂度O(n).

class Solution {
public:
    void duplicateZeros(vector<int>& arr) {
        vector<int> res;
        for(int i = 0; i < arr.size(); ++i)
        {
            res.push_back(arr[i]);
            if(arr[i] == 0)
            {
                res.push_back(0);
            }
        }
        arr.assign(res.begin(), res.begin()+arr.size());
    }
};