题目
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
来源:
https://leetcode-cn.com/problems/move-zeroes/
示例:
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
解答
方法一:
思路如下
- 发现一个0, 把后边的元素依次往前移动,
- 最后一个元素赋值为0
- 每次发现一个0的时候,后边往前移动的元素个数减一,因为最后的都是0,没必要移动了
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int count = 0;
for (int i = 0; i < nums.size() - count;) {
if (nums[i] == 0) {
for (int j = i; j < nums.size() - 1 - count; ++j) {
nums[j] = nums[j + 1];
}
nums[nums.size() - 1 - count] = 0;//移动的最后一个元素
count++;
} else {
i++;
}
}
}
};
方法二:
思路如下:
- 交换0和非0的元素
- 如果0后边不是0 直接交换就行,然后继续往前走
- 如果0后边还是0 那就继续找不是0的交换
所以就变成了发现0的时候寻找后边第一个不是0的元素并且进行交换的操作了。
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for (int i = 0 ; i < nums.size();) {
//找到不是0, 只要发现一个0 就找往后找第一个不是0的
if (nums[i] == 0) {
int left = i;
//往后找第一个不是0的数据
while(true) {
++i;
if (i == nums.size()) {
return;
}
if (nums[i] != 0) {
break;
}
}
//交换
std::swap(nums[left], nums[i]);
//继续从原来的地方往前循环
i = left + 1;
} else {
++i;
}
}
}
};