remove duplicates from sorted array ii（从重复数组中移除重复元素）

类似问题：remove duplicates from sorted list（移除有序链表中的重复元素）

题目描述

Follow up for “Remove Duplicates”:
What if duplicates are allowed at most twice?
For example,
Given sorted array A =[1,1,1,2,2,3],
Your function should return length =5, and A is now[1,1,2,2,3].

题目大意

从重复数组中移除重复元素；
允许最多重复两次。

方法一

思路

常规版本。
设置index索引值，索引数组序号，然后对数组进行遍历；
当有前中后三个连续重复元素时，就执行continue；
否则就把当前值赋值给index下标，并且把index加一。

代码

#include<iostream>
using namespace std;

int removeDuplicates(int A[], int n)
{
    if(n<=2)return n;
    
    int index = 0;
    
    for(int i=0; i<n; i++)
    {
        // 当有三个连续相等的情况
        if(i>0&&i<n-1 && A[i]==A[i+1] && A[i]==A[i-1])
        {
            continue;
        }
        
        A[index++] = A[i];
        
    }
    
    return index;
}

int main()
{
    int a[] = {1, 1, 1, 2, 2, 3, 3, 3};
    
    int n = removeDuplicates(a, 8);
    
    for(int i=0; i<n; i++)
    {
        cout<<a[i]<<' ';
    }
    
    cout<<endl;
    
    return 0;
}

方法二

思路

很灵活的方法，扩展性较强；
如果将occur<2改为occur<3，就变成了允许重复最多三次。

代码

int removeDuplicates_2(int A[], int n) 
{
    if(n<=2)return n;

    int index=2; // 允许重复两次，可以修改为三次

    for(int i=2; i<n; i++)
    {
        if(A[i]!=A[index-2]) // 允许重复两次，可以修改为三次
        {
            A[index++]=A[i];
        }
    }
     
    return index;
}

以上。