题目:
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入:[0,1,0,3,12]
输出:[1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
解答:
简单题:
Approach #1: 把0删了再加在最后面。
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int>::iterator it;
for (it = nums.end() - 1; it >= nums.begin(); it--) {
if (*it == 0) {
nums.erase(it);
nums.push_back(0);
}
}
}
};
Approach #2:
实际上:把所有的0移到最后面 = 把所有不是0的数与此时数组最前面的0交换位置。
具体做法:两个指针,遍历整个数组:
第一个指针之前的数都不是0(一般情况它本身指的数是0);
第一个指针和第二个指针中间的数都是0;
这个时候只要第二个指针指向的数不是0,就可以交换两个指针指向的值。
令:
第一个指针名字叫做beforeItNotZero;
第二个指针名字叫做cur(当前遍历到的位置)
举个栗子:
那么,如果beforeItNotZero不是0怎么办呢?此时cur一定和beforeItNotZero指向同一个位置也就是说,那个时候一定是自己和自己交换了:
再来一个栗子:
下面是代码实现:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for (int beforeItNotZero = 0, cur = 0; cur < nums.size(); cur++) {
if (nums[cur] != 0) {
swap(nums[beforeItNotZero], nums[cur]);
beforeItNotZero++;
}
}
}
};