描述

给定一个增序排列数组 nums ，你需要在 原地 删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

说明：

为什么返回数值是整数，但输出的答案是数组呢？

请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下：

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}
 

示例 1：

输入：nums = [1,1,1,2,2,3]
输出：5, nums = [1,1,2,2,3]
解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 你不需要考虑数组中超出新长度后面的元素。
示例 2：

输入：nums = [0,0,1,1,1,1,2,3,3]
输出：7, nums = [0,0,1,1,2,3,3]
解释：函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。 你不需要考虑数组中超出新长度后面的元素。
 

提示：

0 <= nums.length <= 3 * 104
-104 <= nums[i] <= 104
nums 按递增顺序排列

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

求解

   class Solution {
    public:
        // 第一版，暴力解法，直接遍历元素发现多余2个的直接调用vector接口erase删除
        // 这个办法非常不好，每次删除会导致删除点后面的所有元素都前移拷贝
        int removeDuplicates_1e(vector<int> &nums) {
            const int n = 2; // 最多保留2个重复元素
            int current = nums[0];
            int i = 0;
            int curNum = 0;
            while (i < nums.size()) {
                if (nums[i] == current) {
                    ++curNum;
                    ++i;
                    continue;
                }
                // 出现新元素
                current = nums[i];
                if (curNum > n) {
                    int extraTwoNum = curNum - n; // 超出2的重复数量
                    nums.erase(nums.begin() + i - extraTwoNum, nums.begin() + i); // 删除元素
                    i -= extraTwoNum; // 删除extraTwoNum个元素后，i下标需要前移extraTwoNum位置
                }
                curNum = 1;
                ++i;
            }
            // 处理如[...., x, x, x, x]的场景
            if (curNum > n) {
                int extraTwoNum = curNum - n; // 超出2的重复数量
                nums.erase(nums.begin() + i - extraTwoNum, nums.begin() + i); // 删除元素
            }
            return nums.size();
        }

        // 第二版，使用辅助空间，空间复杂度O(N)，不符合题意
        int removeDuplicates_2e(vector<int> &nums) {
            const int n = nums.size();
            if (nums.size() < 3) {
                return n;
            }
            vector<int> rec;
            rec.reserve(n);
            int cnt = 1;
            int cur = nums[0];
            rec.push_back(nums[0]);
            for (int i = 1; i < n; ++i) {
                if (nums[i] == cur) {
                    if (cnt < 2) {
                        rec.push_back(nums[i]);
                        ++cnt;
                    }
                } else {
                    rec.push_back(nums[i]);
                    cur = nums[i];
                    cnt = 1;
                }
            }
            nums.assign(rec.cbegin(), rec.cend());
            return rec.size();
        }

        // 第三版，双指针,原地处理，空间复杂度O(1)
        int removeDuplicates(vector<int> &nums) {
            const int n = nums.size();
            if (nums.size() < 3) {
                return n;
            }
            int k = 0; // [0...k]保留满足条件的元素
            int cnt = 1;  // 初始情况下已经有一个元素了
            for (int i = 1; i < n; ++i) {
                if (nums[i] == nums[k]) {
                    if (cnt < 2) {
                        nums[++k] = nums[i];
                        ++cnt;
                    }
                } else {
                    nums[++k] = nums[i];
                    cnt = 1;
                }
            }
            return k + 1;
        }
    };