给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/move-zeroes
思路1:
创建两个指针i和j,第一次遍历的时候指针j用来记录当前有多少非0元素。即遍历的时候每遇到一个非0元素就将其往数组左边挪,第一次遍历完后,j指针的下标就指向了最后一个非0元素下标。
第二次遍历的时候,起始位置就从j开始到结束,将剩下的这段区域内的元素全部置为0。
时间复杂度:O(n)
空间复杂度:O(1)
class Solution {
public void moveZeroes(int[] nums) {
if(nums==null) {
return;
}
//第一次遍历的时候,j指针记录非0的个数,只要是非0的统统都赋给nums[j]
int j = 0;
for(int i=0;i<nums.length;++i) {
if(nums[i]!=0) {
nums[j++] = nums[i];
}
}
//非0元素统计完了,剩下的都是0了
//所以第二次遍历把末尾的元素都赋为0即可
for(int i=j;i<nums.length;++i) {
nums[i] = 0;
}
}
}
思路2:
借鉴 快速排序的思想,快速排序首先要确定一个待分割的元素做中间点x,然后把所有小于等于x的元素放到x的左边,大于x的元素放到其右边。
这里我们可以用0当做这个中间点,把不等于0(注意题目没说不能有负数)的放到中间点的左边,等于0的放到其右边。
按照这个思路,代码如下:
可以把j理解为 非零元素的位置、个数。
class Solution {
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;
}
}
}
}
我们发现上述代码存在一个可以改善的地方。对于i==j,对于第一个非零元素,会原地交换,这大可不必。因此改善后,如下:
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) {// #避免了数组开头是非零元素的交换也就是阻止(i==j)时交换。
nums[j] = nums[i];
nums[i] = 0;
}
j++;
}
}
}
Tips:写这道题,看到一个交换的小技巧。一般我们都是用temp临时变量搞得。也可以用异或^ 完成,不用临时变量。
下面^是异或运算。
我们交换A、B
A= A^B
B= A^B
A=A^B
就是利用A=(A^B)^B