给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
1、必须在原数组上操作,不能拷贝额外的数组。
2、 尽量减少操作次数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/move-zeroes
C++
解题思路:
方法1:设置一个参数K,用于保存非0元素。非零元素的范围是[0,k)。
时间复杂度:O(n);
空间复杂度:O(1);
(1)进行一次遍历,如果遇到非0元素,将非0元素放到K的位置上,然后k++。如果是0,不进行任何操作。
(2)将剩余位置补0。
代码如下:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int k = 0; //存放非0元素[0,k)
for(int i = 0; i< nums.size(); i++){
if ( nums[i] ){
nums[k] = nums[i];
k++;
}
}
for(int j = k; j < nums.size(); j++)
nums[j] = 0;
}
};
方法2:设置一个参数K,用于保存非0元素。非零元素的范围是[0,k)。
时间复杂度:O(n);
空间复杂度:O(1);
(1)进行一次遍历,如果遇到非0元素,将非0元素与K的位置交换,然后k++。如果是0,不进行任何操作。
相比于方法1,不再需要进行补0操作。
代码如下:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int k = 0; //存放非0元素[0,k)
for(int i = 0; i< nums.size(); i++){
if ( nums[i] ){
swap(nums[k++] , nums[i]);
}
}
}
};