给定一个数组 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
进阶:你能尽量减少完成的操作次数吗?
题解
暴力的话两层循环可以解决问题,但显然能不暴力就不暴力,这里有个O(n)的方法
反过来思考,直接先把不为0的元素放好,记录不是0的元素的位置i和数目one,遇到一个不为0的数直接放,最后的结果就是非0元素的相对顺序没变,然后把不为0元素后面的都变成0
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int one=0;
for(int i=0;i<nums.size();i++){
if(nums[i]){
nums[one++]=nums[i];
}
}
while(one<nums.size()){
nums[one++]=0;
}
}
};