go语言LeetCode刷题笔记
Hot100系列
第283题移动零
给定一个数组 nums
,编写一个函数将所有 0
移动到数组的末尾,同时保持非零元素的相对顺序。
请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]
示例 2:
输入: nums = [0]
输出: [0]
思路:
使用双指针,一个指针遍历数组,一个工作指针记录不是0的数据。比如说示例1中,数组变化应为
[0,1,0,3,12]//第一个数字为0,不变
[1,1,0,3,12]//第二个数字为1,将index为0的数据改为1
[1,1,0,3,12]//第三个数字为0,不变
[1,3,0,3,12]//第四个数字为3,将index为1的数据改为3
[1,3,12,3,12]//第五个数字为12,将index为2的数据改为12
然后,将数组后续部分(也就是工作指针后面的部分)改为0,数组变化过程如下
[1,3,0,0,12]//将index为3的数据改为0
[1,3,12,3,12]//将index为4的数据改为0,执行完毕
go语言实现代码如下
func moveZeroes(nums []int) {
//设置两个指针,一个p指针在前面记录非零元素,一个i指针遍历数组
p, n := 0, len(nums)
for i := 0; i < n; i++ {
if nums[i] != 0 {
nums[p] = nums[i]
p++
}
}
//将后续元素变为0
for p < n {
nums[p] = 0
p++
}
}