给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
要求:
必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
- 示例
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
思路
使用双指针,左指针指向当前已经处理好的序列的尾部,右指针指向待处理序列的头部。
右指针不断向右移动,每次右指针指向非零数,则将左右指针对应的数交换,同时左指针右移。
- 代码
public void moveZeroes(int[] nums) {
int i=0,j=0;
int n = nums.length;
while (j<n){
if (nums[j]!=0){
swap(nums,i,j);
i++;
}
j++;
}
}
private void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
看评论还有一种写法,如下所示
public void moveZeroes(int[] nums) {
//设置一个index,表示非零数的个数
int index=0;
for (int i = 0; i < nums.length; i++) {
if (nums[i]!=0){
nums[index++]=nums[i];
}
}
for (int i = index; i <nums.length; i++) {
nums[i]=0;
}
}
这种写法看似很巧妙,但其实存在一个问题。例如对于测试用例[0,0,0,0,0,1]
来说,使用双指针方法,只需要交换一次即可。使用方法二的话,需要 N-1
次写入零。而且题目要求是移动零,而不是覆盖。
后记
以后每天都会更新一道力扣题,大部分是简单题,但是包含了很多技巧。希望对大家有所帮助。