题目
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
思路
使用双指针,使一个指针i指向当前遍历到的元素,另一个指针k始终指向一个为0的元素,初始时,i,k均从0开始。
void moveZeroes(vector<int>& nums) {
int k =0;
//遍历找到不为零的数 与为零的数交换
for(int i = 0; i < nums.size(); i++)
{
if(nums[i]) {
swap(nums[i],nums[k]);
k++;
}
}
}
思考
如果不用考虑非零元素的相对顺序,只用将为0的数移到末尾,那么可以怎么实现。使用双指针,一个指向首元素,一个指向末元素,使用前指针进行判断,如果元素是0,则和后指针交换,并将后指针前移,前指针后移;否则只将前指针后移,直到两指针相遇为止。
void moveZeroes(vector<int>& nums) {
for(int i = 0, k = nums.size() -1 ; i < k ; i++)
{
if(nums[i] == 0){swap(nums[i],nums[k]); k--;}
}
}