需求:
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入:[0, 1, 0, 3, 12]
输出:[1, 3, 12, 0, 0]
要求:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
废话少说,直接上代码:
class Solution:
def moveZeroes(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
j = 0
for i in range(len(nums)):
if nums[i] != 0:
nums[j], nums[i] = nums[i], nums[j]
j += 1
总结:
代码中使用j来记录0所在的位置,for循环里面交换的始终是0和非零元素。
例:我们输入nums = [1, 0, 5, 7, 2, 0, 9], 看一下每一步的交换过程,红色箭头代表每次遍历i的位置,红色元素代表当次遍历中发生交换的两个元素,如图:
遍历结束,所有的零也都移动到列表的后面,非零元素的相对位置也没有变动。