给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
func removeElement(_ nums: inout [Int], _ val: Int) -> Int { nums.removeAll { $0 == val } return nums.count }
右指针 right 指向当前将要处理的元素,左指针 left 指向下一个将要赋值的位置。
如果右指针指向的元素不等于 val,它一定是输出数组的一个元素,我们就将右指针指向的元素复制到左指针位置,然后将左右指针同时右移;
如果右指针指向的元素等于 val,它不能在输出数组里,此时左指针不动,右指针右移一位。
func removeElement(_ nums: inout [Int], _ val: Int) -> Int { var lp = 0, rp = 0 while rp < nums.count { if nums[rp] == val { rp += 1 } else { nums[lp] = nums[rp] rp += 1 lp += 1 } } return lp }
依然使用双指针,两个指针初始时分别位于数组的首尾,向中间移动遍历该序列。
func removeElement(_ nums: inout [Int], _ val: Int) -> Int { guard nums.count > 0 else { return 0 } var lp = 0, rp = nums.count - 1 while lp < rp { if nums[lp] == val { nums[lp] = nums[rp] rp -= 1 } else { lp += 1 } } // 当 lp 与 rp 重合, rp 指向的元素等于 val if nums[rp] == val { rp -= 1 } return rp + 1 }
更简的写法, 当 while 的条件加上 = 时, 正好可以覆盖 当 lp 与 rp 重合, rp 指向的元素等于 val 的这种情况, 感兴趣的话, 可以思考下原因
func removeElement(_ nums: inout [Int], _ val: Int) -> Int { guard nums.count > 0 else { return 0 } var lp = 0, rp = nums.count - 1 while lp <= rp { if nums[lp] == val { nums[lp] = nums[rp] rp -= 1 } else { lp += 1 } } return rp + 1 }