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)