计蒜客 移除数组中的重复元素

移除数组中的重复元素

给定一个升序排列的数组，去掉重复的数，并返回新的数组的长度。

例如：

   数组A = {1, 1, 2}，你的函数应该返回长度2，新数组为{1, 2}

要求：

   不能新开数组分配额外的空间。即常数空间限制。

提示：

   输入一个整数n，以及其对应的数组A[n]，输出新数组长度

样例输入

5
0 0 1 1 2

样例输出

3

#include<iostream>
#include<vector>
#define MAX 100000
using namespace std;
int main()
{
    int n, ans;
    int a[MAX];
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);

    ans = n;
    for (int i = 0; i<n - 1; i++)
    {
        if (a[i] == a[i + 1])
            ans--;
    }

    printf("%d", ans);
}