Python, LintCode, 172. 删除元素

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
        elem_index = [i for i,val in enumerate(A) if val == elem]
        for i in range(len(elem_index)):
            A.remove(elem)
        return len(A)

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
        elem_count = A.count(elem)
        for i in range(elem_count):
            A.remove(elem)
        return len(A)