移除元素

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

function removeElement(nums,val) {
    for(let i = nums.length - 1;i > -1;i-- ){
        if(nums[i] == val){
            nums.splice(i,1)
        }
    }
    return nums.length
}

快慢指针

/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
const removeElement = function (nums = [0, 1, 2, 2, 3, 0, 4, 2], val = 2) {
    let slow = -1
    for (let i = 0; i < nums.length; i++) {
        const v = nums[i]
        if (v === val) {
            if (slow === -1)
                slow = i
        } else {
            if (slow !== -1) {
                [nums[slow], nums[i]] = [nums[i], nums[slow]]
                i = slow
                slow = -1
            }
        }
    }
    return slow !== -1 ? nums.slice(0, slow).length : nums.length
};

Leecode提交通过

posted @ 2020-07-14 18:29  671_MrSix  阅读(84)  评论(0编辑  收藏  举报