给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
尽量减少操作次数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/move-zeroes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
使用双指针 left 和 right :
- left:已经排好序的尾部。
- right:没排序的头部。
right 一直右移,若 right 遇到非零数字,就与 left 交换并且 left 右移。比如[100120]
:
- 第一次交换:
[10(left)01(right)20]
→[110(left)02(right)0]
- 第二次交换:
[110(left)02(right)0]
→[1120(left)00(right)]
left 和 right 之间的数字为 0 ,left 左边为非零数字。
class Solution:
def moveZeroes(self, nums: List[int]) -> None:
"""
Do not return anything, modify nums in-place instead.
"""
left = right = 0
n = len(nums)
while right < n:
if nums[right] != 0:
nums[left], nums[right] = nums[right], nums[left]
left += 1
right += 1