计蒜客 排序后的数组删除重复数字

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

分析：因为是排好序的，扫描一次数组就够了，每当同一个数字出现次数超过2次，长度就减1；

#include <iostream>
#include <vector>
using namespace std;

int fun(vector<int> v){
    int sz = v.size(), temp = 1, len = sz;
    for(int i = 1; i < sz; i++){
        if(v[i] == v[i - 1])
            temp++;
        else
            temp = 1;
        if(temp > 2)
            len--;
    }
    return len;
}

int main(){
    int n;
    cin >> n;
    vector<int> v(n);
    for(int i = 0; i < n; i++)
        cin >> v[i];
    cout << fun(v) << endl;
    return 0;
}