题目
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums =[0,1,0,3,12]
输出:[1,3,12,0,0]
示例 2:
输入: nums =[0]
输出:[0]
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
第一遍
自己写了一份,创建一个数组,for循环提取出非零数,在末尾补零。
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int> result;
int i=0;
for(auto iter = nums.begin();iter != nums.end();++iter){
if(*iter){
result.push_back(*iter);
}
else{
++i;
}
}
for(;i >0 ;--i){
result.push_back(0);
}
nums=result;
}
};
第二遍
看了官方分享的代码,用swap函数替换,left为零的位置,自己根据代码写了一份
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int i=0;
for(auto iter = nums.begin();iter != nums.end();++iter){
if(*iter)
swap(*iter , *(nums.begin() + (i++) ));
}
}
};