LeetCode:Remove Duplicates from Sorted Array II

很常见的题目，容易错的地方是没有判断超出的个数以及如果没有超过，则需要将count回归初始值

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n<3)//如果n小于3，不存在重复三次的数字，直接返回原长度
             return n;
        int i,j;
        int count;
        int k;
        count=1;
        for(i=0;i<n;i++)
        {
            if(i<n-1&&A[i]==A[i+1])
            {
                count++;
                if(count==3)//重复超过了2个，则需要判断超出的个数
                {
                   j=i+1;
                   k=0;//超出个数
                   while(A[j]==A[i]&&j<n)
                   {
                       j++;
                       k++;
                   }
                   for(j=i+1;j+k<n;j++)
                   {
                       A[j]=A[j+k];
                   }
                   n=n-k;
                   count=1;
                }
            }
            else//如果为超出，就将重新计算
               count=1;
        }
        return n;
    }
};