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;                }                       }           };