LeetCode Remove Duplicates from Sorted Array II

题目大意：给出一个排序后的数组，去除数组中的重复元素且最多允许有两个元素相同，最后返回处理后的数组长度，且数组是整理后的。

算法思想：当数组长度小于3时不用整理数组，直接返回数组的长度；当数组长度大于等于3时，用pre记录前驱元素，flag标记是否重复一次，p记录新数组的末端坐标，然后扫描整个数组，相邻两个元素比较，如果相同且flag=0则将比较的元素放入到新数组中，flag=1,反之，如果量元素相同则continue,若不相同将比较元素放入到新数组中flag=0。

代码如下：

class Solution {
public:
     int removeDuplicates(int A[], int n) {
            int p;//记录新数组的末端
            int pre;//前趋变量
            int flag;//标记元素重复几次
            if(n==0||n==1||n==2){
                return n;
            }
            else{
                pre=A[0];
                flag=0;
                p=1;
                for(int i=1;i<n;i++){
                     if(pre==A[i]&&!flag){
                         A[p++]=A[i]; 
                         flag=1;     
                    }
                    else{
                         if(pre==A[i]) continue;
                        else{
                            pre=A[i];
                            A[p++]=A[i];
                            flag=0;
                        }
                    }  
                }
                return p;    
            }                  
     }           
};