计蒜客第22题---排序后的数组删除重复数字

给定升序排序的数组，如果数组有2个或2个以上相同的数字，去掉他们，直到剩下2个为止。
例如：
数组A[] = [1,1,1,2,2,3]
你的算法需要返回新数组的长度5,
此时A为[1,1,2,2,3].
格式：第一行输入一个数字n，第二行输入A[n],
最后输出新数组的长度。
样例1

输入：
6
1 1 1 1 3 3
输出：
4

#include <iostream>
using namespace std;
int sortRemove(int n, int A[])
{
    int k = 0;//超过2的次数
    int num = 1;//记录每个元素出现的次数
    for (int j = 0; j<n; j++)
    {
        if (A[j] == A[j + 1])
            num++;
        else
            num = 1;
        if (num>2)
            k++;
    }
    return n - k;
}

int main()
{
    int n, m, result;
    cin >> n;
    int *A = new int[n];
    for (int i = 0; i<n; i++)
    {
        cin >> m;
        A[i] = m;
    }
    result = sortRemove(n, A);
    cout << result << endl;
    delete [ ]A;
    return 0;
}