给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
进阶:你能尽量减少完成的操作次数吗?
如果只操作原数组,那有个思路,就是用双指针,一个指向第一个0,一个指向0后边的第一个非0,将他们交换位置,再往后遍历。
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int zero=0,nonzero=1;
int size=nums.size();
while(true)
{
//找第一个0,当后边所有数字都是0,return
while(nums[zero]!=0&&zero<size-1)
{
zero++;
}
if(zero==size-1) return;
//找0后边的第一个非0,当后边全是0时,return
while((nums[nonzero]==0||nonzero<=zero)&&nonzero<size-1)
{
nonzero++;
}
if(nums[nonzero]==0) return;
else
{
nums[zero]=nums[nonzero];
nums[nonzero]=0;
}
}
}
};
然后看了下官方题解,思路是一样的,但是代码实现比我精炼很多
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int n = nums.size(), left = 0, right = 0;
while (right < n) {
if (nums[right]) {
swap(nums[left], nums[right]);
left++;
}
right++;
}
}
};