题目:
给你一个长度固定的整数数组 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]
思路:借助栈,将arr数组中元素进栈,遇见0要连续多压入1个,当栈元素等于arr的长度时停止压入元素,然后将栈中元素出栈,进入arr中,最后将arr中元素反转(因为栈中元素先进后出)
代码:第一版自己用vector<int> 来模拟队列的功能(先进先出)
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
int len = arr.size();
//用tem模拟队列
vector<int> tem;
//初始化tem
tem.push_back(arr[0]);
for (int i = 0; i < len; i++) {
//tem中第一个元素是要插入i位置上的数
if (tem[0] != 0) {
//不为0直接将tem首元素插入i位置
arr[i] = tem[0];
//当到arr数组最后一个元素时,后面的数就都不要了
if ((i + 1) < len) {
//将下一个元素压入
tem.push_back(arr[i + 1]);
}
//删除首元素
tem.erase(tem.begin());
continue;
}
//当等于0时要同时将两个元素压入,因为arr在i,i+1位置都要插入0
if ((i + 1) < len) {
tem.push_back(arr[i + 1]);
}
if ((i + 2) < len) {
tem.push_back(arr[i + 2]);
}
arr[i] = 0;
if ((i + 1) < len) {
arr[i + 1] = 0;
}
tem.erase(tem.begin());
i++;
}
}
};
第二版,用栈实现
class Solution {
public:
void duplicateZeros(vector<int>& arr) {
stack<int> st;
//控制栈中元素结束插入,因为栈中元素是最后插入到arr中的元素
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());
}
};
第一版是内存动态的分配,第二版是栈的大小为arr.size();前者内存好点,但时间慢