题目描述
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例一:
给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
示例二:
给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。 注意这五个元素可为任意顺序。 你不需要考虑数组中超出新长度后面的元素。
思路
1.这道题可以使用快慢指针来解决。
2.每次都将快指针的值赋值给慢指针,然后判断当前附的值是否为目标值,若不是目标值,则慢指针移动,否则慢指针不动。
3.最后我们将慢指针的数值返回即可。
Java代码实现
public int removeElement(int[] nums, int val) { int i,j=0; for (i = 0; i < nums.length; i++) { nums[j] = nums[i]; if(nums[i] != val){ j++; } } return j; }
Golang代码实现
func removeElement(nums []int, val int) int { sort.Ints(nums) res := 0 for i:=0;i<len(nums);i++{ if nums[i] != val { nums[res] = nums[i] res++ } } return res }