27. 移除元素 | 双指针 | TypeScript

题目

27. 移除元素

解题思路

思路简单，但是边界条件一不小心就会出错

• 元素的顺序可以改变，使用双指针交换元素即可

实现

• 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;
};