我们就下面这道简单算法题加深对双指针理解与运用
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums =[0,1,0,3,12]
输出:[1,3,12,0,0]
示例 2:
输入: nums =[0]
输出:[0]
题目给的要求很简单,只需要把零放在数组的最后就可以了。
不妨有很多种实现的方法,比如先提取所有非零数字再在最后添加零。还有甚至可以简单的调用库函数只用一行代码解决:
fill(remove(nums.begin(),nums.end(),0),nums.end(),0);
但在这里我们使用双指针如下解决:
class Solution {
public:
void moveZeroes(vector<int>& nums) {
int i = 0, j = 0;
int len = nums.size();
while(i < len && j < len){
if(nums[j] == 0)
j ++;
else{
swap(nums[i], nums[j]);
i ++; //下一处(若有)仍然是0
j ++;
}
}
}
};
虽然是一道简单的算法题,但我觉得这个方法才是最能体现算法意义以及乐趣的。
该算法使用两个指针,i
和j
,来遍历vector。以下是算法的关键思想:
- 初始化两个指针,
i
和j
,指向vector的开始。 - 将
i
和j
都初始化为0。 - 使用一个while循环来遍历vector,直到
i
小于vector的长度。 - 在循环中,检查
j
指向的元素是否为0。如果是0,则递增j
以移动到下一个元素。 - 如果
j
指向的元素不是0,则交换它与i
指向的元素。然后,递增i
和j
以移动到下一个元素。这里的想法是,由于j
指向的元素不是0,它肯定位于其正确的位置,所以交换它与i
指向的元素,并移动两个指针。 - 循环继续,直到所有非零元素都移动到它们正确的位置。
该算法确保在O(n)O(n)时间内将零移动到vector的末尾,其中nn是vector的长度。