LintCode Python 简单级题目 100.删除排序数组中的重复数字 101.删除排序数组中的重复数字II

题目100描述：

给定一个排序数组，在原数组中删除重复出现的数字，使得每个元素只出现一次，并且返回新的数组的长度。

不要使用额外的数组空间，必须在原地没有额外空间的条件下完成。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出数组A =[1,1,2]，你的函数应该返回长度2，此时A=[1,2]。

标签 

两根指针 数组 脸书

 

题目101描述：

跟进“删除重复数字”：

如果可以允许出现两次重复将如何处理？

您在真实的面试中是否遇到过这个题？ 

Yes

样例

 

标签 

两根指针 数组 脸书

题目分析：

给定的数组是排序数组；

循环数组，找到下一个元素A[i+1]等于A[i]，就删除A[i+1]，

注意的是循环次数需要 -1，避免索引超限。

 

允许重复一次，则直接检查元素A[i+2]是否等于A[i]即可，

因为如果A[i+2]等于A[i]，那么A[i+1]也等于A[i]，

此时删除A[i+2]即可。

当然，循环次数需要 -2

源码：

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

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