# 1089. 复写零

力扣原题链接(点我直达) (opens new window)

给你一个长度固定的整数数组 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++;
    }
  }
}
1
2
3
4
5
6
7
8
9

执行用时 :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());
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

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

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