给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
解法一(效率差)
双重循环
外层:遍历元素
内层:往后找到第一个不是0的元素
交换0和之后第一个不是0的元素
后面全是0,提前退出循环
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for (int i = 0; i < nums.size(); i++) {
if (nums[i] == 0) {
int k = -1;
for (int j = i+1; j < nums.size(); j++) {
if (nums[j] != 0) {
k = j;
break;
}
}
if (k == -1)
return;
int t = nums[i];
nums[i] = nums[k];
nums[k] = t;
}
}
}
};
解法二(常规解法,必须掌握)
记录z为写入下标,非0值移动到该下标处,0跳过,遍历一遍后非0值全部移动到z之前了,这时候将[z,n)下标的置0
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int z = 0;
int n = nums.size();
for (int i = 0; i < n; i++) {
if (nums[i] != 0) {
nums[z++] = nums[i];
}
}
while (z < n) {
nums[z++] = 0;
}
}
};