前端算法-数组篇(三)-移除元素

1.题目

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

移除元素

2.题解

2.1 暴力解法

两层for循环，一个for循环遍历数组元素 ，第二个for循环更新数组。时间复杂度是O(n^2)。

2.2 快慢指针

• 快指针：寻找新数组的元素 ，新数组就是不含有目标元素的数组
• 慢指针：指向更新 新数组下标的位置

时间复杂度：O(n)，空间复杂度：O(1)

3. 代码

3.1 暴力解法

function removeElement(nums: number[], val: number): number {
    
    
    // 暴力解法，双for循环，第一个for循环遍历元素 ，第二个循环更新数组
    for(let i: number = 0; i < nums.length; i++) {
    
    
        const temp = nums[i];
        if (temp === val) {
    
    
            for(let j: number = i + 1; j < nums.length; j++) {
    
    
                nums[j - 1] = nums[j];
            }
            // 都前移了一位，所以下标也要更新
            i--;
            nums.pop();
        }
    }
    return nums.length;
};

3.2 快慢指针

function removeElement(nums: number[], val: number): number {
    
    
    // 快慢指针
    let fastIndex: number = 0, slowIndex: number = 0;
    while(fastIndex < nums.length) {
    
    
        if (nums[fastIndex] !== val) {
    
    
            nums[slowIndex++] = nums[fastIndex];
        }
        fastIndex++;
    }
    return slowIndex;
};

4. 参考

代码随想录-移除元素