题目
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums =[0,1,0,3,12]
输出:[1,3,12,0,0]
示例 2:
输入: nums =[0]
输出:[0]
思路
思路是利用双指针,因为要保持数组中非零元素的相对顺序,所以都从第一个元素往后遍历
左指针指向零元素,右指针指向左指针后面的第一个非零元素,然后交换。
左指针的问题好解决,可以向后遍历停在第一个0的位置
关键问题就是:
如何保持右指针指向左指针后面的第一个非零元素
假如左指针已经就位,右指针指向的为非零元素,那么需要交换。右指针指向的也是零,那就需要继续向后走
第一版代码
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int>::iterator first = nums.begin(), last = nums.begin();
while(last != nums.end() && first != nums.end())
{
if(*first ==0 && *last != 0)
{
int tmp = *last;
*last = *first;
*first = tmp;
first++;
}
if(*first)
first++;
if(*last==0)
last++;
}
}
};
但是这样会出现first > last的情况,反而把零元素放到前面了。
之后尝试弥补这个问题,结果越写越乱。最后还是看了官方题解,爆了!!
修改后的代码:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
vector<int>::iterator first = nums.begin(), last = nums.begin();
while(last != nums.end())
{
if(*last != 0)
{
int tmp = *last;
*last = *first;
*first = tmp;
first++;
}
last++;
}
}
};
意思是
将first看作待交换的第一个元素
这样last不断寻找非零元素,再把它们放到first记录的最终位置
这样就忽视了要把0放到最后,问题变成了将所有非零元素换到前面,而不是寻找左指针右边的第一个非零元素。
难绷,本来想用类似链表的方式,记录开始first和结束last的位置,让first移动,直到遇见last。然后遇见0就先删除,再插入末尾。
没想到在vector中,这个last并不是像指针那样不动的,在删除后last指向的内容竟然变了