给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
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.
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
Note:
- You must do this in-place without making a copy of the array.
- Minimize the total number of operations.
解题思路:
千万不要被题目局限了思维!题目让把所有0移到末尾,如果你的思路是遇零与末尾数字交换位置,然后还需要把非零数字排序,那么就被带偏了。
换个思路,把非 0 数字前移,不去管数字 0。
定义两个指针:指针 i 直接遍历数组遇到非 0 数字把该数字赋值给指针 j 所在的索引,索引 j 自增 1,i继续遍历。
这样遍历完之后,数组索引从0到 j 之间的数值即为所求得保持非零元素的相对顺序
,而 j 之后的数值只需要全部赋值 0 即可。
Java:
class Solution {
public void moveZeroes(int[] nums) {
int numsLen = nums.length;
if (numsLen < 1) return;//数组长度小于一直接返回
int j = 0;
for (int i = 0; i < numsLen; i++) {//遍历数组
if (nums[i] != 0) {//如果该数不为0
nums[j++] = nums[i];//赋值给索引j
}
}
while (j < numsLen) nums[j++] = 0;//把从j到末尾所有数字赋值0
}
}
Python3:
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
if len(nums) < 1:
return
j = 0
for num in nums:
if num != 0:
nums[j] = num
j += 1
for i in range(j, len(nums)):
nums[i] = 0
如果题目不限制在原数组上操作,用python一行可解决:
nums = [i *for* i in nums *if* i != 0]+[i *for* i in nums *if* i == 0]
公众号: 爱写bug(ID:iCodeBugs)