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

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

 

跟进“删除重复数字”：

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

在：lintcode100 删除排序数组中的重复数字 的基础上进行改进。

class Solution {
public:
    /*
     * @param nums: An ineger array
     * @return: An integer
     */
    int removeDuplicates(vector<int> &nums) {
        // write your code here
        int len = nums.size();
        if (nums.empty()) return 0;

        int index = 1;
        int c = 1;  //c是关键
        for (int i = 1; i < len; ++i) {
            if (nums[i] != nums[i - 1]) {
                nums[index] = nums[i];
                c = 1;
                index++;
            } else if ((nums[i] == nums[i - 1]) && c < 2){
                nums[index] = nums[i];
                index++;
                c++;
            }
        }
        return index;
    }
};