[Leetcode] Remove duplicates from sorted array ii 从已排序的数组中删除重复元素

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].

 题意：可以保留一个重复的元素。

思路：第一种是和Remove duplicates from sorted array类似的思想。隔一个比较一次，拿A =[1,1,1,2,2,3]举例子，其对比的过程如下

 

这里可以将第6行、第10行的2改为3，则至多重复的数字出现三次，代码为：

class Solution 
{
public:
    int removeDuplicates(int A[], int n) 
    {
        if(n<=2)    return n;
        int lo=2;
        for(int i=2;i<n;++i)
        {
            if(A[i] !=A[lo-2])
                A[lo++]=A[i];
        }
        return lo;
    }
};

方法二：使用计数器，当两者不相等计数器重置，将A[i]赋值给A[lo]的下一位；相等时但是count不超过2，将A[i]赋值给A[lo]的下一位，其余，仅计数器++，这样就可以成功将不等的情况后的，出现第二个和A[i]相等的情况时，将其也赋值给前面的元素。

class Solution {
public:
    int removeDuplicates(int A[], int n) 
    {
        int lo=0;
        int count=0;
        if(n<3) return n;

        for(int i=1;i<n;i++)
        {
            if(A[lo]==A[i])
            {
                count++;
                if(count<2)
                    A[++lo]=A[i];
            }
            else
            {
                A[++lo]=A[i];
                count=0;
            }
        }
        return lo+1;
    }
};