lintcode 172. Remove Element

难度:

1. Description

lintcode 172. Remove Element_第1张图片
172. Remove Element

2. Solution

  • python
    用list的remove函数
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
        for i in range(A.count(elem)):
            A.remove(elem)
        return len(A)

不用list的remove函数

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
        j = len(A)-1
        for i in range(len(A)-1, -1, -1):
            if A[i] == elem:
                A[j], A[i] = A[i], A[j]
                j -= 1 
        return j+1 

3. Reference

  1. https://www.lintcode.com/problem/remove-element/description

你可能感兴趣的:(lintcode 172. Remove Element)