思路1:将非零元素拿出来放到数组前面去,后面的用0来填补
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int> nonZeroElements;
for(int i=0;i<nums.size();i++){
if(nums[i])
nonZeroElements.push_back(nums[i]);
}
for(int i=0;i<nonZeroElements.size();i++){
nums[i]=nonZeroElements[i];
}
for(int i=nonZeroElements.size();i<nums.size();i++){
nums[i]=0;
}
}
};
利用了额外的空间,不是原地算法,因此需要进一步的优化
思路2:改进:在遍历的过程中,将非零元素直接往前移,用k保存遍历过的非零元素
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int k = 0; //[0,k)中为非零元素
for (int i = 0; i < nums.size(); i++) {
if (nums[i])
nums[k++] = nums[i];
}
for (int i = k ; i < nums.size(); i++)
nums[i] = 0;
}
};
思路3:将非零元素与0交换位置
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int k = 0; //[0,k)中为非零元素
for (int i = 0; i < nums.size(); i++) {
if (nums[i])
swap(nums[k++], nums[i]);
}
}
};