要求
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
思路
方法一:双指针,两次遍历
我们创建两个指针 ‘i’ 和 ‘j’,第一次遍历的时候指针 ‘j’ 用来记录当前有多少非0元素。即遍历的时候每遇到一个非0元素就将其往数组左边挪,第一次遍历完后,‘j’ 指针的下标就指向了最后一个非0元素下标。
第二次遍历的时候,起始位置就从 ‘j’ 开始到结束,将剩下的这段区域内的元素全部置为0。
public class LeetCode283 {
public void moveZeroes(int[] nums) {
if (nums == null){
return;
}
//第一次遍历时,j遇到元素为‘0’的下标,则停下,等待i遍历到非‘0’元素的下标,则把nums[i]赋值给nums[j]
int j = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] != 0){
nums[j] = nums[i];
j++;
}
}
//非0的元素统计完了,第二次遍历把末尾的元素都赋为0
for (int i = j; i < nums.length; i++) {
nums[i] = 0;
}
}
}
时间复杂度: O(n)
空间复杂度: O(1)
方法二:一次遍历
这里参考了快速排序的思想,快速排序首先要确定一个待分割的元素做中间点x,然后把所有小于等于x的元素放到x的左边,大于x的元素放到其右边。
这里我们可以用0当做这个中间点,把不等于0(注意题目没说不能有负数)的放到中间点的左边,等于0的放到其右边。这的中间点就是0本身,所以实现起来比快速排序简单很多,我们使用两个指针i和j,只要nums[i]!=0,我们就交换nums[i]和nums[j]
public void moveZeroes(int[] nums) {
if(nums==null) {
return;
}
//两个指针i和j
int j = 0;
for(int i=0;i<nums.length;i++) {
//当前元素!=0,就把其交换到左边,等于0的交换到右边
if(nums[i]!=0) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j++] = tmp;
}
}
}
下面是一个评论区大佬的解法,对上面的代码进行了优化,再#1处
public void moveZeroes(int[] nums) {
int length;
if (nums == null || (length = nums.length) == 0) {
return;
}
int j = 0;
for (int i = 0; i < length; i++) {
if (nums[i] != 0) {
//这里比较的是元素下标奥,跟元素的正负没有关系奥,我看下面评论的人有误区
if (i > j) {// #1
nums[j] = nums[i];
nums[i] = 0;
}
j++;
}
}
}
时间复杂度: O(n)
空间复杂度: O(1)