Leetcode 80. 删除排序数组中的重复项 ii 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比较类似，区别只是这里元素可以重复出现至多两次，而不是一次。其实也比较简单，只需要维护一个counter，当counter是2时，就直接跳过即可，否则说明元素出现次数没有超，继续放入结果数组，若遇到新元素则重置counter。总体算法只需要扫描一次数组，所以时间上是O(n)，空间上只需要维护一个index和counter，所以是O(1)。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(A==NULL||n<1) return 0;
        int count=0;
        int res=0;
        for(int i=0;i<n;i++){
            if(i>0&&A[i]==A[i-1]){
                count++;
            }else
                count=0;
            if(count<2){
                A[res++]=A[i];
            }
        }
        return res;
    }
};

 或者直接比较j-1和j-2

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n=nums.size();
        int i=0,j=0;
        for(i=0;i<n;++i)
        {
            if(j>1&&nums[i]==nums[j-1]&&nums[i]==nums[j-2])
                continue;
            nums[j++]=nums[i];
        }
        return j;
    }
};