leetcode 日经贴，python code -Remove Element

Remove Element

class Solution:    # @param    A       a list of integers    # @param    elem    an integer, value need to be removed    # @return an integer    def removeElement(self, A, elem):        i, j = 0, len(A) - 1        while i <= j:            if A[i] == elem:                A[i], A[j] = A[j], A[i]                j -= 1            else: i += 1        return i