Remove Duplicates from Sorted Array II

一、问题描述

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array nums = [1,1,1,2,2,3],

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

二、思路

去除数组中重复数大于2的元素，并将数组最后长度返回。

最关键的地方在于判断多余的变量flag，由于是有序数组，当下一个元素和当前元素不同时，flag变量重新赋值为0。还有vector的erase函数必须要注意参数类型是string::iterator的类型，而且删除元素后指向下一个元素，所以我们用--i将指针回到原位置。

三、代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n = nums.size();
        int flag = 0;
        for(int i = 1; i < n; ++i){
            if(nums[i] == nums[i - 1]){
                flag++;
                if(flag > 1){
                    --n;
                    nums.erase(nums.begin() + i);
                    --i;
                }
            }else{
                flag = 0;
            }
        }
        return n;
    }
};