代码随想录-数组-移除元素

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

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

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

示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

你不需要考虑数组中超出新长度后面的元素

思路

暴力解法

从数组中查找指定值的元素,很容易想到遍历依次去比较元素的值,查找到目标元素后要移除该元素,并且要求原地修改输入数组,则不能直接删除该元素,而是需要该元素后的所有元素依次替代原有值,从而得到移除元素的效果。

因此需要两层for循环,外层for循环用于查找要被删除的元素,内层for循环用于将后面元素都往前移一位以覆盖要被删除的元素,更新数组。最后返回新数组的长度。因此我们可以将数组长度设为一个单独的变量,方便对其进行操作。

js 复制代码
function remove(nums,target){
    let len = nums.length
    // 外层for循环,查询目标元素
    for(let i =0; i < len; i++){
        if(nums[i] == target){
            // 查询到要删除的元素,开始遍历后面元素,依次覆盖之前元素,更新数组
            // 要使用下标为j++的元素覆盖下标为j的元素,j++为最后一个元素下标时为len-1;则j的最大值应该小于len-1
            for(let j = i; j < len - 1; j++){
                // 使用后一个元素覆盖前面元素的值
                nums[j] == nums[j++]
            }
            // 一个元素被移除(覆盖),则数组总长度应减一
            len --;
            // 目标元素被后面元素所覆盖,当下一次查询时应该还从当前元素开始查询,需要判断覆盖之后的元素是否为要删除的目标元素,所以i要减1,确保for循环i++后下次循环开始时i还指向被覆盖的元素
            i --;
        }
    }
    // 每次移除元素len都会减1,则len为移除所有目标元素后的数组长度
    return len;
}

双指针法(快慢指针)

快慢指针法:使用一个快指针一个慢指针在一个for循环中完成两个for循环的工作。

快指针:进行遍历,指向当前遍历到的元素的下标,用快指针判断是否为要移除的目标元素

慢指针:在遍历过程中指向新数组当前要更新的元素的下标。

js 复制代码
function remove(nums,target){
    let slow = 0;
    for(let fast = 0; fast < nums.length; fast ++){
        // 如果当前元素为要移除的目标元素,则slow指针不移动,fast指针后移,等待下一个fast指向的元素不为目标元素时,将fast指向的元素的值赋给slow指针指向的元素,并将slow后移一步
        if(nums[fast] !== target){
            nums[slow++] = nums[fast]
        }
    }
    // 每次更新元素后(将当然不是目标元素的值赋给slow指向的元素),slow都会被+1,对应更新后数组的长度,可以直接返回
    return slow;
}
相关推荐
tqs_123452 分钟前
分sheet写入excel
开发语言·python·算法
西望云天4 分钟前
基础组合计数(三道例题)
数据结构·算法·icpc
San309 分钟前
JavaScript 流程控制与数组操作全解析:从条件判断到数据高效处理
javascript·面试·代码规范
温宇飞12 分钟前
CSS 中如何处理空白字符
前端
dengzhenyue13 分钟前
矩形碰撞检测
开发语言·前端·javascript
llq_35018 分钟前
为什么 npm view yarn version 显示 1.22.22?
前端
aesthetician31 分钟前
ReactFlow:构建交互式节点流程图的完全指南
前端·流程图·react
neo_dowithless35 分钟前
多语言维护太痛苦?我自研了一个翻译自动化 CLI 工具
前端·ai编程
小徐_233336 分钟前
老乡鸡也开源?我用 Trae SOLO 做了个像老乡鸡那样做饭小程序!
前端·trae
前端伪大叔1 小时前
第13篇:🎯 如何精准控制买入卖出价格?entry/exit\_pricing 实战配置
javascript·python