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 <= 100000 <= 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%的用户