数据结构与算法--27.移除元素

27.移除元素

题目描述

题目链接：题目链接

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

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

思路1–暴力求解

首先这题如果考虑暴力解法的话，正常逻辑也是相当简单，就是用for循环遍历，找到元素然后对比找到的元素与目标值是否一样，相同则删除，不同则不删除。但是如果按这种方法运行出的结果是不对的，感觉逻辑也没有啥问题，但是就是不对。原因在于我们如果正顺序删除代码的话，每次删除后，数组的长度会改变同时后面的会向前进一步**（涉及底层，大概应该是这样的，才疏学浅，可能理解不深，相互学习哈**），基于此问题，我们考虑从数组的最后一个元素逐步向前搜索直到找到第一个元素，遇到需要删除的元素也是后面处理过的元素向前进一步，并不会影响未处理的元素，这我们用从后往前删除查找的方法就可以成功避开这个问题，

错误代码及结果

def removeElement(nums, val):
    """
    :type nums: List[int]
    :type val: int
    :rtype: int
    # """
    for i in nums:
        if i == val:
            nums.remove(i)
    return len(nums),nums


# nums = [1]
nums = [0,1,2,2,3,0,4,2]

x,y = removeElement(nums,2)
print(x)
print("================")
print(y)

可以发现结果的最后一个2没有被删除，但是感觉逻辑又没有问题。

代码演示

class Solution(object):
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        n = len(nums) - 1
        for i in range(n,-1,-1):
            if(nums[i] == val):
                nums.remove(nums[i])
        return len(nums)

结果分析

思路2–双指针

双指针法（快慢指针法）： 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。

我觉得双指针就是用一个for循环，取两个下标，根据条件用快指针对应的数据给慢指针对应数组的元素赋值。结果输出慢指针对应的数组。

代码演示

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

结果演示

参考

https://leetcode-cn.com/problems/remove-element
代码随想录