LintCode_100 删除排序数组中的重复数字 ||

题目

跟进“删除重复数字”：

如果可以允许出现两次重复将如何处理？

样例

给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

C++代码

int removeDuplicates(vector<int> &nums) {
        // write your code here
        if(nums.size() == 0) return 0;
        int i,j;
        int count = 1;
        for(i = 1,j = 1; i < nums.size(); ++i)
        {
            if(nums[i] != nums[j-1])
            {
                nums[j++] = nums[i];
                count = 0;
            }
            else if(count < 2 && nums[i] == nums[j-1])
            {
                nums[j++] = nums[i];
            }
            count++;
        }
}