leetcode 【 Remove Element 】python 实现

题目:

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

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

 

代码 : oj测试通过 Runtime: 43 ms

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

 

思路:

顺序便利数组元素；遇到值等于elem的，就与尾部元素交换；整个过程中维护尾部交换元素的指针位置