2/100移除元素

问题描述:给你一个数组 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。

这个问题有两个解法,初看题目“不要使用额外的数组空间”就可以想到的是暴力解,实现暴力解发现时间复杂度为O(n^2);对于“不需要考虑数组中超出新长度后面的元素”数组问题提高算法效率尝试使用双指针,设计好双指针之间的逻辑可以高效解决问题。

暴力法:

使用两层for循环,一个for循环遍历数组元素 ,第二个for循环更新数组,定义一个整型记录数组长度,每删除一个元素长度减1。

int removeElement(vector<int>& nums, int val) {

        int size = nums.size();

        for (int i = 0; i < size; i++) {

            if (nums[i] == val) { // 发现需要移除的元素,就将数组集体向前移动一位

                for (int j = i + 1; j < size; j++) {

                    nums[j - 1] = nums[j];

                }

                i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位

                size--; // 此时数组的大小-1

            }

        }

        return size;

}

  • 时间复杂度:O(n^2)
  • 空间复杂度:O(1)

双指针法:

    • 定义一快(fast)一慢(slow)两个指针,同时从数组下标为0的元素出发
    • 判断fast指向的值是不是目标值,不是则fast++,slow++
    • 遇到目标值快指针fast多走一步,然后将慢指针slow指向的值赋值为快指针指向的值,重复第②步
    • 快指针fast走完数组结束
  • 快指针:寻找新数组的元素,新数组就是不含有目标元素的数组
  • 慢指针:指向更新 新数组 下标的位置

这种方法又称之为快慢指针法

int removeElement(vector<int>& nums, int val) {

        int slowIndex = 0;

        for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {

            if (val != nums[fastIndex]) {

                nums[slowIndex++] = nums[fastIndex];

            }

        }

        return slowIndex;

}
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值