地址 https://leetcode-cn.com/problems/move-zeroes/
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
解法
另开个数组 或者额外开多个空间存放额外的0 肯定是可以的,但是不符合题意。
可以考虑使用双指针,找到符合交换条件的0 和非0 进行替换。
条件如下
1 l指针,以当前l为起点 找到第一个为0的位置
2 r指针,以当前r为起点 找到第一个非0的位置 且r在l之后
(初始的位置有这个问题 比如 1 0 2,应该定位2的位置而不是1 的位置)
3 r在l之后 且位置合法 进行替换。
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int l = 0; int r = 0;
while (l < nums.size() && r < nums.size()) {
//从当前位置开始 找到第一个为0 的位置
while (l < nums.size() && nums[l] != 0) l++;
if (l >= nums.size()) { break; }
//从当前位置开始 找到第一个非0的位置
while ( (r < nums.size() && nums[r] == 0) || r<=l) r++;
if (r >= nums.size()) { break; }
if (l < r) { swap(nums[l],nums[r]); }
l++; r++;
}
}
};