给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
- 示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
我的方案
思路
将这道题分成三个步骤来做:
- 数0的个数
- 将非零元素赋值到集合中
- 添0,并将集合赋值到数组中
代码
public void moveZeroes(int[] nums){
int zeroCount = 0;
List<Integer> list = new ArrayList<Integer>();
for(int i =0;i<nums.length;i++) {
if(nums[i]==0) {
zeroCount++;
}
}
for(int i = 0; i<nums.length;i++) {
if(nums[i]!=0) {
list.add(nums[i]);
}
}
while(zeroCount--!=0) {
list.add(0);
}
for(int i=0;i<nums.length;i++) {
nums[i] = list.get(i);
}
}
官方方案一
思路
像这种带筛选性质的数组题,可以考虑用快慢指针来做。特别是这道题需要筛选出来的是同一个元素:0
算法
快指针向前寻找,找到一个非零元素来赋值给慢指针,这时快慢指针同时前进。最后根据非零元素总数与原本数组总数的差值来补充0
代码
public void moveZeroes(int[] nums){
int lastZeroNum = 0;
for(int i = 0;i<nums.length;i++) {
if(nums[i]!=0) {
nums[lastZeroNum++] = nums[i];
}
}
for(int i =lastZeroNum;i<nums.length;i++) {
nums[i] = 0;
}
}
官方方案二
思路
上一个算法属于局部优化,对于像这样的数组【0,0,0,0,0…1】,在写下唯一一个非零元素后,需要再写入0 n次。针对这种情况优化,发现可以不写入0,而是用快指针指向的非零元素与慢指针(也是指向第一个零元素的指针)进行交换。
算法
快慢指针齐头并进,直到慢指针找到0停下,快指针依旧前进,直到找到非0停下,然后两者继续齐头并进。
代码
public void moveZeroes(int[] nums){
int firstZero = 0;
for(int cur =0;cur<nums.length;cur++) {
//如果当前快指针元素为0,快指针前进一步
//如果当前快指针不是0,交换两者所指元素,并且快慢指针都前进一步
//这样慢指针会在第一个零处停下,而快指针继续前进直到下一个非零。
if(nums[cur]!=0) {
int temp = nums[cur];
nums[cur] = nums[firstZero];
nums[firstZero] = temp;
firstZero++;
}
}