[Lintcode] 172. Remove Element

最新推荐文章于 2019-07-25 11:51:28 发布

siriusli_

最新推荐文章于 2019-07-25 11:51:28 发布

阅读量136

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode

本文链接：https://blog.youkuaiyun.com/siriusli1981/article/details/86025270

lintcode 专栏收录该内容

27 篇文章

订阅专栏

本文提供了一种解决LeetCode上172.RemoveElement问题的有效方法，通过两种不同的Python代码实现，详细解释了如何从数组中移除指定元素并返回新的长度。第一种方法使用while循环和pop函数，第二种方法则采用双指针技巧，提高了效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

172. Remove Element

Description

Given an array and a value, remove all occurrences of that value in place and return the new length.

The order of elements can be changed, and the elements after the new length don’t matter.

Example
Given an array [0,4,4,0,0,2,4,4], value=4

return 4 and front four elements of the array is [0,0,0,2]

我的代码：

class Solution:
    """
    @param: A: A list of integers
    @param: elem: An integer
    @return: The new length after remove
    """
    def removeElement(self, A, elem):
        # write your code here
        f = 0
        while f != len(A):
            if A[f] == elem:
                A.pop(f)
            else:
                f += 1
        return f

方法二：

class Solution:
    def removeElement(self, A, elem):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """        
        l = len(A)
        head = 0
        tail = len(A) - 1
        if tail == 0:
            if A[0] == elem:
                return 0
            else:
                return 1
        while head < tail:
            while A[head] != elem:
                head += 1
                if head==len(A):
                    return len(A)
            while A[tail] == elem:
                tail -= 1
                if tail<0:
                    break
            if head > tail:
                break
            tmp = A[head]
            A[head] = A[tail]
            A[tail] = tmp
        return head