移动零
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
提示:
1 <= nums.length <= 104
-231 <= nums[i] <= 231 - 1
进阶:你能尽量减少完成的操作次数吗?
解答:
-
用两个索引指针,一个指向元素0,一个指向非0元素
-
然后依次交换
-
不能跳针
-
错误的想法,这种无法覆盖[1,0]
public class MoveZeroesClass283 {
public static void moveZeroes(int[] nums) {
if (nums == null || nums.length == 1) return;
int zIndex = 0;//0指针
int nzIndex = 0;//非0指针
while (zIndex < nums.length && nzIndex < nums.length) {
if (nums[zIndex] != 0) {
zIndex++;
continue;
}
if (nums[nzIndex] == 0) {
nzIndex++;
continue;
}
//交换
int temp = nums[zIndex];
nums[zIndex] = nums[nzIndex];
nums[nzIndex] = temp;
// zIndex = nzIndex; //不能跳针
}
return;
}
}
- 参考快排想法,将0作为中间节点,遍历一遍,时间复杂度O(n)
public class MoveZeroClass {
public void moveZeroes(int[] nums) {
if (nums == null || nums.length == 1) return;
int zIndex = 0;//0指针
int nzIndex = 0;//非0指针
//参考快排想法,将0作为中间节点
while (zIndex < nums.length ) {
if (nums[zIndex] != 0) {
//交换
int temp = nums[zIndex];
nums[zIndex] = nums[nzIndex];
nums[nzIndex++] = temp;
}
zIndex++;
}
}
public static void main(String[] args) {
int[] nums = {5,1,3,12,0};
MoveZeroClass solution = new MoveZeroClass();
solution.moveZeroes(nums);
for (int i = 0 ; i < nums.length; i++) System.out.println(nums[i]);
}
}
- 更容易的实现,遍历两边,时间复杂度依然O(n)
//简单的实现
public void moveZeroes(int[] nums) {
if (nums == null || nums.length == 1) return;
int j = 0;//非0指针
//将所有j指向所有非0元素
for (int i = 0; i < nums.length; i++) {
if (nums[i] != 0) {
nums[j++] = nums[i];
}
}
//将剩下的元素赋值为0元素
for (int i = j; i < nums.length; i++) {
nums[i] = 0;
}
}