刪除有序數組中的重複元素，返回數組的新長度,但允許最大重複次數爲2

/************************************************************************************
**description:刪除有序數組中的重複元素，返回數組的新長度,但允許最大重複次數爲2
**            如：Input:a={2,2,2,15}  Output:len=3,a={2,2,15}
************************************************************************************/

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

//方法一：用index標識新數組的索引，遍歷數組元素，當前元素與新數組的倒數第二個元素不同時，將其加入新數組
//時間複雜度O(n),空間複雜度O(1)
int removeDuplicates2_1(vector<int>& arr)
{
    if (arr.size() <= 2)
        return arr.size();

    int index = 1;
    for (int i = 2; i < arr.size(); i++)
    {
        if (arr[i] != arr[index - 1])
            arr[++index] = arr[i];
    }
    return index + 1;
}

//方法二：遍歷數組元素，當前元素與它前一個和後一個都相等時，該數不加入
//時間複雜度O(n),空間複雜度O(1)
int removeDuplicates2_2(vector<int>& arr)
{
    if (arr.size() <= 2)
        return arr.size();

    int index = 0;
    for (int i = 0; i < arr.size(); i++)
    {
        if (i > 0 && i < arr.size() - 1 && arr[i] == arr[i - 1] && arr[i] == arr[i + 1])
            continue;
        arr[index++] = arr[i];
    }
    return index;
}