Python|計蒜客——排序後的數組刪除重複數字

給定升序排序的數組，如果數組有 2 個或 2個以上相同的數字，去掉他們，直到剩下 2個爲止。

例如：

數組 A[] = [1,1,1,2,2,3]

你的算法需要返回新數組的長度 5，

此時 A爲 [1,1,2,2,3]

輸入格式

第一行輸入一個不超過 200的數字 n，第二行輸入 A[n]。

輸出格式

輸出新數組的長度。

樣例輸入

6 1 1 1 1 3 3
樣例輸出

4

代碼：

n = int(input())
A = [int(i) for i in (input().split())]
l = len(A)
c = 0
for i in range(l-1):
    if A[i]==A[i+1]:
        c+=1
        if c>=2:
            l-=1
        else:
            pass
    else:
        c=0
print(l)

沒有太多要說的