题目链接
https://leetcode.cn/problems/remove-element/
题解
题解一(Java)
可以使用双指针的方式,定义两个指针i, j开始的时候分别指向数组的第一个元素和最后一个元素
1、每次比较i下标的元素是否是目标元素,如果是,和j下标的元素交换,j向前移动,i不移动,为什么i不移动,因为如果j也是目标元素的话,岂不是要被跳过去了吗
2、什么时候结束循环,当i指针和j指针相遇,则整个数组的所有元素就都被筛查了一遍
3、当代码结束,i和j相遇后,j表示的是无效元素的首元素,所以j-1下标其实才是有效的下标,但是题目求的是长度,下标+1是长度,所以直接返回j,并且循环条件时候也是i≤j,如果题目要求返回的是下标,直接i即可
public int removeElement(int[] nums, int val) {
// 双指针
int i = 0;
int j = nums.length - 1;
while (i <= j) {
if (nums[i] == val) {
swap(nums, i, j--);
} else {
i++;
}
}
return i;
}
public void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = nums[i];
}
题解二(Go)
func removeElement(nums []int, val int) int {
if nums == nil || len(nums) == 0 {
return 0
}
left, right := 0, len(nums)-1
for left <= right {
if nums[left] == val {
// 交换
nums[left], nums[right] = nums[right], nums[left]
right--
} else {
left++
}
}
return left
}