题目如下
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [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++;
}
}
};
复杂度分析
时间复杂度:O(n),其中 n 为序列长度。每个位置至多被遍历两次。
空间复杂度:O(1)。只需要常数的空间存放若干变量。
代码2
class Solution
{
public:
void moveZeroes(vector<int>& nums)
{
int indexNow = 0;
int indexNum = 0;
int m = nums.size();
while (indexNum < m)
{
if (nums[indexNum] != 0)
{
nums[indexNow++] = nums[indexNum];
}
++indexNum;
}
for (int i = indexNow; i < m; i++)
{
nums[i] = 0;
}
}
};