算法练习——排序后的数组删除重复数字

给定升序排序的数组，如果数组有 $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

#include<cstdio>
#include<algorithm>
using namespace std;
static const int MAX=200;

int main()
{
    int n,A[MAX],result=0;
    int p=0,r=0;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&A[i]);
    while(p<n&&r<n)
    {
        while(A[p]==A[r]&&r<n)
        {
            r++;
        }
        if(p-r>2||p-r==2) result+=2;
        else result++;
        p=r;
    }
    printf("%d",result);
}