要求:
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入:[0,1,0,3,12]
输出:[1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
思路:
可以使用erase()函数先删除掉vector容器中元素0,同时记录下数组中所有0元素的个数,然后再利用push_back()将0添加到末尾处。
程序:
void moveZeroes(vector<int>& nums) {
int flag = 0;
for(vector<int>::iterator iter = nums.begin(); iter != nums.end();){
if (*iter == 0){
iter = nums.erase(iter); //返回值指向删除元素的下一个元素
flag++;
}
else
iter++;
}
for(int i = 0;i < flag;i ++){
nums.push_back(0);
}
}
注意:
使用erase()函数一定要注意,详细如下博主所总结,https://blog.csdn.net/a1232345/article/details/49746921,在此感谢该博主。如下程序:
for(int i = 0;i < nums.size();i++){
if(nums[i]==0){
nums.erase(nums.begin()+i);
nums.push_back(0);
}
}
发现这种方法不适合两个0连在一起的情况。如[0,0,1]输出结果为[0,1,0]。思考是否可以采用remove()函数进行操作?