给定一个数组 nums, 编写一个函数将所有 0 移动到它的末尾,同时保持非零元素的相对顺序。
例如, 定义 nums = [0, 1, 0, 3, 12],调用函数之后, nums 应为 [1, 3, 12, 0, 0]。
注意事项:
必须在原数组上操作,不要为一个新数组分配额外空间。
尽量减少操作总数。
思路:增加标志位(j)记录从头开始的非零元素后面的位置,循环数组,当元素非零时,交换nums[i]和nums[j],即将找到的非零元素移动到当前非零元素串后面。
代码如下:
class Solution {
func moveZeroes(_ nums: inout [Int]) {
let count = nums.count
var j = 0
for i in 0..<count {
if nums[i] != 0 {
nums.swapAt(i, j)
j += 1
}
}
}
}
//这里交换用的是Swift提供的 swapAt(i: Int, j: Int) 函数
//当然交换可以自己写,我是真的有点懒