LeetCode Remove Duplicates from Sorted Array II 删除整型数组中的重复元素并返回剩下元素个数2

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        int *s=&A[0],*e=&A[0];    //s指向“连续数字”的第一个，e往后遍历相同的
        int i,flag=0,j=n;
        for(i=1;i<n;i++){
            e++;
            if(*s!=*e||(*s==*e&&flag==0) ){        //新的数字，直接加复制到s处
                if(*s==*e)
                    flag=1;
                else
                    flag=0;
                s++;
                *s=*e;
            }
            else
                j--;
        }
        return j;
    }

};

题意：提供一个整型数组，和该数组内有n个元素，数组已经有序。相同的数字不允许>2个，要么1个，要么2个。修改原数组，并返回剩下的元素个数。

思路：增加一个标志flag=0，遇到多个相同的数字时，第一个正常处理，在第2个数字的处理时将标志改变一下为非零，那么后面相同的就可以去掉了。