题目大意
给定一个数组nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
思路分析
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
根据示例就是把0元素移动到数组的末尾,还要保障非0元素的相对有序。
这个方法可以有两种解法:
- 遍历2遍列表:第一遍遍历,首先取一个全局下标
j
,如果遇到非0值则j+1
, 依次覆盖原始数据,遍历一遍以后j
刚好指到最后一个非0元素的下标。第二遍遍历,起始位置就从 j 开始到结束,将剩下的下标值赋值为0即可。 - 遍历1遍列表:参考快速排序的思想,快速排序首先要确定一个待分割的元素做中间点 x,然后把所有小于等于 x 的元素放到 x 的左边,大于 x 的元素放到其右边。我们可以用 0 当做这个中间点,把不等于 0(注意题目没说不能有负数)的放到中间点的左边,等于 0 的放到其右边。我们使用两个指针 i 和 j,只要
nums[i]!=0
,我们就交换nums[i]
和nums[j]
代码示例
Java版
// 解法1
class Solution {
public void moveZeroes(int[] nums) {
int j = 0;
int len = nums.length;
for(int i=0; i<len; i++){
if(nums[i] != 0)
nums[j++] = nums[i];
}
for(int i=j; i<len; i++){
nums[i] = 0;
}
}
}
// 解法2
class Solution {
public void moveZeroes(int[] nums) {
//两个指针i和j
int j = 0;
for(int i=0; i<nums.length; i++) {
//当前元素!=0,就把其交换到左边,等于0的交换到右边
if(nums[i]!=0) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j++] = tmp;
}
}
}
}
Python版本
# 解法1
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
j = 0
for num in nums:
if num != 0:
nums[j] = num
j += 1
for i in range(j, len(nums)):
nums[i] = 0
# 解法2
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
j = 0
for i in range(len(nums)):
if nums[i] != 0:
nums[j],nums[i] = nums[i],nums[j]
j += 1