【leetcode】Remove Duplicates from Sorted Array 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的算法，但是其中的A[start++]=A[i]的运算在这个允许重复的数组中，将会导致数字的覆盖，因此我的解决方法就是加了一个数组B，来判断重复，然后在A中进行数据的删减。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n==0)return 0;
        if(n==1)return 1;
        if(n==2)return 2;
        int answer=n;
        int *B=(int*)malloc((n)*sizeof(B));//数组的空间分配
        for(int i=0;i<n;i++)
        {
            B[i]=A[i];
        }
        int start=2;
        for(int i=2;i<n;i++)
        {
            if(B[i]==B[i-2])
            {
                answer--;
            }
            else
            {
                A[start]=B[i];
                start++;
            }
        }
        return answer;
    }
};

但是觉得还是可以不用加数组，用双指针的方法来做，在网上找到了如下的方法（用count来记录重复的次数）：

class Solution {
public int removeDuplicates(int[] A) {  
    if(A==null || A.length==0)  
        return 0;  
    int idx = 0;  
    int count = 0;  
    for(int i=0;i<A.length;i++)  
    {  
        if(i>0 && A[i]==A[i-1])  
        {  
            count++;  
            if(count>=3)  
             continue;  
        }  
        else
        {  
            count = 1;  
        }  
        A[idx++]=A[i];  
    }  
    return idx;  
}