给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入:[0,1,0,3,12] 输出:[1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
【思路】这道题题目默认让你移动零,所以可以用两个指针进行交换,找到一个零和它后面的一个非零,交换即可。
【代码】
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int i=0;
int j=1;
while(j<nums.size()&&i<nums.size()){
while(nums[i]!=0){
if(i==nums.size()-1)
break;
++i;
}
j = nums.size()-1==i?i:i+1;
while(nums[j]==0){
if(j==nums.size()-1)
break;
++j;
}
swap(nums[i++],nums[j++]);
}
}
};
这样写显得十分笨重,于是有下面的优化,我们让两个指针同时从0出发,只考虑快指针,当快指针指向非零元素时,我们需要交换快和慢指针指向的元素,然后推进两个指针。如果它是零元素,我们只是推进快指针,这样慢指针自然就指向了0。代码很简单如下:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for(int slow=0,fast=0;fast<nums.size();++fast){
if(nums[fast]!=0)
swap(nums[slow++],nums[fast]);
}
}
};
当然这道题也有其他解法:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int j=0;
for(int i=0;i<nums.size();++i){
if(nums[i]!=0)
nums[j++]=nums[i];
}
for(;j<nums.size();++j){
nums[j]=0;
}
}
};