Given an array nums, write a function to move all 0’s to the end of it while maintaining the relative order of the non-zero elements.
Example:
Input: [0,1,0,3,12]
Output: [1,3,12,0,0]
题目要求把0移到最后,而且不能用一个新数组保存。
比较笨的方法是冒泡法,每个0都冒泡调到最后,但是复杂度会比较高,这里用一个时间复杂度O(N),空间复杂度O(1)的方法。
思路:
只需要从左到右把0和非0对换,这样0后面的非0元素会占据原来0的位置,就达到了0被移到右边的效果。
比如上面的例子:
0,1,0,3,12
左边第一个0和第一个非0(1)对换,得到
1,0,0,3,12
左边第一个0和第一个非0(3)对换,得到
1,3,0,0,12
左边第一个0和第一个非0(12)对换,得到
1,3,12,0,0
也就得到了结果
那么需要两个指针,一个指向左边起第一个0(leftZeroAt),一个指向左边起第一个非0(cur)
刚开始它们都指向第0个元素,
当cur指向的元素为0时,cur++
cur非0时就和leftZeroAt指向的元素对换,然后cur和leftZeroAt都右移。
这时肯定会想要是leftZeroAt指向的元素不是0怎么办,
举个例子:
1,0,0,3,12
开始的时候cur和leftZeroAt都指向1,因为cur指向非0,所以和leftZeroAt指向的元素对换,自己和自己换相当于不变。然后cur和leftZeroAt都右移,指向0,因为cur指向0时右移,这时cur就和leftZeroAt分开了,leftZeroAt停在了0,而cur右移寻找非0。
也就是说cur和leftZeroAt的起点是一样的,如果指向非0,它们会同时右移,如果指向0,leftZeroAt会停在0,而cur去寻找非0。这时候,如果
(1) leftZeroAt和cur相邻
比如:
1, 0, 2, 3
leftZeroAt: 0, cur: 2
对换后,leftZeroAt右移指向对换后的0,cur右移指向3
(2) leftZeroAt和cur不相邻
1,0,0,2, 3
leftZeroAt: 0, cur: 2
对换后,leftZeroAt右移指向下一个0,cur右移指向3
public void moveZeroes(int[] nums) {
if (nums == null || nums.length == 0) {
return;
}
for(int leftZeroAt = 0, cur = 0; cur < nums.length; cur++) {
if (nums[cur] != 0) {
swap(nums, leftZeroAt, cur);
leftZeroAt++;
}
}
}
public void swap(int[] nums, int leftZeroAt, int cur) {
if (leftZeroAt == cur) {
return;
}
int tmp = nums[leftZeroAt];
nums[leftZeroAt] = nums[cur];
nums[cur] = tmp;
}