172. Remove Element

題目

https://www.lintcode.com/problem/remove-element/description?_from=ladder&&fromId=2

實現

1. 定義一個 result_length 作爲不相等元素的指針
2. 遍歷數組，如果元素和 target 不相等，那麼 A[result_length] 和 A[index] 的元素進行交換
3. 最終返回 result_length - 1

代碼

class Solution:
    """
    @param: A: A list of integers
    @param: elem: An integer
    @return: The new length after remove
    """
    def removeElement(self, A, elem):
        result_length = 0

        for index in range(len(A)):
            if A[index] != elem:
                temp = A[index]
                A[index] = A[result_length]
                A[result_length] = temp

                result_length += 1

        return result_length