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

第一步 比较耗时

void duplicateZeros(vector<int>& arr) {
  for (int i=0; i<arr.size(); i++){
    if (arr[i] == 0){
      arr.insert(arr.begin()+i, 0);
      arr.pop_back();//这步很关键
      i++;
    }
  }
}

执行用时 :48 ms, 在所有 C++ 提交中击败了56.70%的用户

内存消耗 :9.5 MB, 在所有 C++ 提交中击败了100.00%的用户

第二种 借助栈,减少时间

void duplicateZeros(vector<int>& arr) {
    stack<int> st;
    int temp = 0;
    for (int i = 0; i < arr.size(); ++i) {
        st.push(arr[i]);
        ++temp;
        if (arr[i] == 0) {
            if (temp == arr.size())
                break;
            ++temp;
            st.push(0);
        }
        if (temp == arr.size())
            break;
    }
    arr.clear();
    while (!st.empty()) {
        arr.push_back(st.top());
        st.pop();
    }
    reverse(arr.begin(), arr.end());
}

执行用时 :28 ms, 在所有 C++ 提交中击败了90.69%的用户

内存消耗 :9.5 MB, 在所有 C++ 提交中击败了100.00%的用户