LintCode 删除排序数组中的重复数字 II

删除排序数组中的重复数字 II

跟进删除排序数组中的重复数字：
如果可以允许出现两次重复将如何处理？
样例
给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

这题和删除重复的数据是一样的，唯一不同的是我们需要保存一个计数器来记录当前重复数据的个数

code :

class Solution {
public:
    /**
     * @param A: a list of integers
     * @return : return an integer
     */
    int removeDuplicates(vector<int> &nums) {
        // write your code here
        int i = 0;
        int count = 1; 
        if (nums.size() == 0)
            return 0;
        for (int j = 1; j < nums.size(); ++j) {
            if (nums[i] == nums[j] && count < 2) {
                ++count;
                nums[++i] = nums[j];
            } else if (nums[i] != nums[j]) {
                count = 1;
                nums[++i] = nums[j] ;
            }
        }
        return i+1;
    }
};