给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
思路:先统计容器中0
的个数,删掉所有的零,并在结尾补零
class Solution
{
public:
void moveZeroes(vector<int>& nums)
{
int count_0 = 0; //统计0的个数
for (vector<int>::iterator it = nums.begin(); it != nums.end(); ) //删掉0
{
if (*it == 0)
{
it = nums.erase(it);
++count_0;
}
else
++it;
}
for (int i = 0; i < count_0; ++i) //结尾补0
nums.push_back(0);
}
};
这里在刚开始用了一种迭代器失效的方法,erase
以后其实改变了容器元素的排列顺序,遍历下标i
其实就失效了,再根据i
来遍历是错的。erase
在删除一个迭代器指向的元素后,会返回已删除元素的下一个相邻元素的迭代器,可以用erase
的返回值来重置迭代器it
,注意此时的for循环的++操作应在循环体内进行。
//for (int i = 0; i < nums.size(); i++)
// {
// if (nums[i] == 0)
// {
// nums.erase(nums.begin() + i); 等同于迭代器失效
// nums.push_back(0);
// }
// }
参考:如何正确删除vector中的元素
class Solution {
public:
void moveZeroes(vector<int>& nums) {
for(int i = 0, j = 0; i < nums.size(); i ++) {
if(nums[i] != 0) {swap(nums[j], nums[i]); ++j;}
}
}
};
快指针i
指向最新的元素,慢指针j
指向【已确认的最后一个非零元素】之后的第一个0
。i
每次指向新的元素,如果nums[i]
是非零数,将nums[i]
与j
处的元素(其实是0)交换位置
参考:leetcode回答
还有更好地优化方法在官方回答中
参考:官方回答
输入: [0,1,0,3,12]
输入: [1,1,0,3,12]
这两个测试用例主要是体会一下快慢指针的更新,手算一下就明白了
输入: [0,1,0,3,12]
输入: [1,1,0,3,12]
这两个测试用例主要是体会一下快慢指针的更新,手算一下就明白了
在原数组上进行删除某个特定值的元素时,使用erase
方法要注意迭代器失效的问题,erase
方法会返回已删除元素的下一个元素的迭代器,可以用erase
返回值来重置迭代器