python|計蒜客——移除數組中的重複元素

給定一個升序排列的數組，去掉重複的數，並輸出新的數組的長度。

例如：數組 A = {1, 1, 2}A={1,1,2}，你的程序應該輸出 22 即新數組的長度，新數組爲 {1, 2}{1,2}。

要求：不能新開數組分配額外的空間，即常數空間限制。

輸入格式

輸入一個整數 n(1 \leq n \leq 1000)n(1≤n≤1000)。

接下來一行 nn 個整數 A_i(-1000 \leq A_i \leq 1000)A ​i ​​ (−1000≤A ​i ​​
≤1000)，表示數組 AA 中的每個元素。

輸出格式

輸出一個整數，表示新數組長度。

樣例輸入

5
0 0 1 1 2
樣例輸出

3

代碼：

n = int(input())
A = [int(i) for i in input().split()]

for i in range(n-1):
    if A[i]==A[i+1]:
        n-=1
print(n)