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)