1.2移除元素(LC27)

本文讲述了如何使用双指针法(快慢指针)在查找并移除列表中特定元素的过程中提高效率。作者分析了代码调试过程,指出`return`语句的位置问题,并给出了正确的代码实现。同时讨论了何时适合使用内嵌函数作为解题步骤。
摘要由CSDN通过智能技术生成

算法原理

多余的元素,删掉不就得了。

但是数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖。

这个题目暴力的解法就是两层for循环,一个for循环遍历数组元素 ,第二个for循环更新数组。

// 时间复杂度:O(n^2)
// 空间复杂度:O(1)
class Solution {
public:
    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;

    }
};

​​​​​​​

使用双指针法(快慢指针法),可以节省时间复杂度。通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。

定义快慢指针

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

两个指针都指向一个数组,fast的任务是找到符合要求的新数组元素,slow则对应 新数组元素的下标。

代码调试过程

列表名是nums,调用元素时要用nums[slow],中括号!!!

修改后的代码:

原因:return slow在while循环里面,当第一次while循环时,3=3,所以slow=0,return回空值。由于return slow在while循环里面,return一次以后就打断了循环。

当在 while 循环内部执行 `return` 语句时,会立即中断循环并返回相应的值。

所以,`return slow`语句应该位于循环的外部,以便在遍历完整个列表后返回结果。

继续修改:

原因:fast指针+1的位置不对。如果 `nums[fast]` 不等于指定的值 `val`,会执行 `nums[slow] = nums[fast]``slow += 1` 和 `fast += 1`,这会导致 `slow` 和 `fast` 同时递增。这样会导致处理元素的跳过,因为下一个元素将被放置到 `nums[slow]` 的位置,但 `slow` 的值已经递增了。这将导致结果列表中的元素数量少于预期,或者在处理元素时出现错误。

  • `slow += 1` 在 `if nums[fast] != val` 的条件下递增。这是因为只有当 `nums[fast]` 不等于指定的值 `val` 时,才需要将该元素移动到 `nums` 的 `slow` 索引位置。因此,只有在这个条件下,我们才会递增 `slow` 的值,以确保下一个非指定元素能够被正确地放置。

  • `fast += 1` 则是在每次迭代中都递增。它的目的是在遍历列表时逐个检查元素,并将 `fast` 递增以处理下一个元素。无论当前元素是否等于指定的值 `val`,都需要递增 `fast`,以便在下一次循环中检查下一个元素。

正确的代码

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        slow=0
        fast=0
        size=len(nums)
        while fast<size:
            if nums[fast]!=val:
                nums[slow]=nums[fast]
                slow+=1
            fast+=1            
        return slow

使用enumerate函数,它允许你在遍历列表或其他可迭代对象时,同时获取元素的索引和值:

#双指针法
#快指针:找目标元素
#慢指针:更新index
#enumerate函数,它允许你在遍历列表或其他可迭代对象时,同时获取元素的索引和值
# nums = [10, 20, 30, 40, 50]

# # 使用 enumerate 获取索引和值
# for index, value in enumerate(nums):
#     print(f"Index: {index}, Value: {value}")

# Index: 0, Value: 10
# Index: 1, Value: 20
# Index: 2, Value: 30
# Index: 3, Value: 40
# Index: 4, Value: 50

#本质是双指针法:通过一个简单的循环和双指针技巧正确地移除了数组中的指定元素。
class Solution(object):
    def remove(self, nums, val):
        new = 0
        for r, x in enumerate(nums):
            if (x != val):
                nums[new] = x
                new += 1           
        return new

2024.8.26三刷

class Solution(object):
    def removeElement(self, nums, val):
        new = 0
        for x in nums:
            if (x != val):
                nums[new] = x
                new += 1
        return new

其他

其实这个代码也可以直接调用remove、erase等内嵌函数,但是这样只要一句话就能实现了,显然不是面试所需。

什么时候可以用内嵌函数?

当该函数只是解题的一个步骤,并且我们知道该函数的时间、空间复杂度。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值