LintCode:删除排序数组中的重复数字 II

LintCode:删除排序数组中的重复数字 II

class Solution:
    """
    @param A: a list of integers
    @return an integer
    """
    def removeDuplicates(self, A):
        # write your code here
        n = len(A)
        if n == 0 or n == 1:
            return n
        i = 0
        j = n-2
        while i < j:
            if A[i] == A[i+1] and A[i] == A[i+2]:
                A.remove(A[i])
                j -= 1
            else:
                i += 1
        return len(A)