【数组】11:移除元素


Leetcode 27: 移除元素(python)

注:本人文章中有时有来自于各大佬的经验总结,仅供自身复习及学习,并无商业用途,若有时忘记标注作者及出处还请提醒,敬请见谅。

我的思路

由于题目要求原地移除元素,所以我们不能使用额外的辅助空间。因此,我们可以很直观地想到将所有值等于val的元素依次移到数组尾部。
具体的执行思路为:用一个变量count记录当前找到的值等于val的元素的个数,用n-count记录元素被挪到数组尾部的位置,如果当前尾部元素的值也为val,则将count加一,再往前一个位置进行寻找,直到找到尾部中下标大于i (这是为了避免将val又换回到数组头部)且值不等于val的位置为止,然后将两个元素进行交换。

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        count = 0
        n = len(nums)
        for i in range(n):
            if (nums[i] == val) and (i < (n - count)):
                count += 1
                # 寻找到当前元素被交换到尾部的位置
                # 如果当前尾部元素的值也为val,则向前一个位置进行寻找
                while (nums[n - count] == val) and (n - count) > i:
                    count += 1
                # 如果i位置后面的元素全部都为val
                if (n - count)== i:break
                # 交换元素
                else:
                    temp = nums[n-count]
                    nums[n - count] = nums[i]
                    nums[i] = temp

        return (n - count)

大佬思路:双指针

看题解的时候发现了双指针的妙用:
用一个指针a遍历数组,用另一个指针记录数组中最靠前的值为val的元素的位置。当a遍历到值不为val的元素时,将a和b对应位置上的元素交换位置,之后b自增1 。

图源代码随想录:
图源代码随想录

具体代码为:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:		
		a = 0
        b = 0

        while a < len(nums):
            if nums[a] != val:
                nums[b] = nums[a]
                b += 1
            a += 1

        return b

作者:flying_du
链接:https://leetcode-cn.com/problems/remove-element/solution/python-shuang-zhi-zhen-da-fa-hao-a-quan-guo-zui-ca/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值