题目
解题思路
思路简单,但是边界条件一不小心就会出错
- 元素的顺序可以改变,使用双指针交换元素即可
实现
- left=0,rightnums.length-1
- 如果nums[left] === val,nums[left] = nums[right–],再次检查nums[left]
- 当 left <= right 的情况下进行操作,如果最后nums[left] === val,则返回left,否则返回left+1
- 注意判断的是最后一个元素!
代码
function removeElement(nums: number[], val: number): number {
if (nums.length === 0) return 0;
let i = 0, j = nums.length - 1;
while (i <= j) {
while (i <= j && nums[i] === val) {
nums[i] = nums[j--]; // 防止交换后的元素还是等于val
}
i++;
}
// return nums[i] === val ? i - 1 : i; // 这种形式判断的是新长度最后一个元素的后一个元素,不正确
i--; // 因为最后i++; 此时的i就是实际的长度,需要减去1,判断最后一个元素
return nums[i] === val ? i : i + 1;
};