80. Remove Duplicates from Sorted Array II（删除排序数组中的重复项 II）解法（C++ & 注释）

1. 题目描述

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

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

示例 1:

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

示例 2:

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

说明:

为什么返回数值是整数，但输出的答案是数组呢?
请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝 int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。 for (int
i = 0; i < len; i++) {
print(nums[i]);
}

题目链接：中文题目；英文题目；

2. 计数（Count）

2.1 解题思路

笔者觉得这题做出来不难，但是想用最优的方法做出来还是比较难想的。首先，我们先来看看比较普遍的思路，最后深入到最优的方法上面，循序渐进。

一个比较直接的思路为，这部分参考代码：2.2.1 详细版：

1. 从序号（idx） = 0，开始计数（count），相同数字count++；
2. 不同数字，如果count <= 2，则重置count = 1，继续向后计数，如果count > 2，则把后面的数字移动到前面重复数字的位置处；
3. 重复上面步骤，直到遍历完整个数组；

还不是很懂的童鞋，可以参考下面的图片：

这里我们需要注意一个很重要的地方：

回溯的序号距离和减少的数字个数都为count - 2，即nums[idx - diff] = nums[idx] 和 len -= diff

利用这个特点，接下来我们将会把上面的思路简化到极致。其实上面方法的主要思想是把个数小于3的数字移动到前面idx - (count - 2)的地方，而且题目给的条件：nums[i] <= nums[i +1]，我们到这里只是用了等于的条件，还没有使用大于的条件。

以这两个点为切入点，我们再来观察一下下面两个例子：

没有数字的个数大于2：[1,1,2,2,3,4]

这种情况下除了idx = 0 或 1，剩下的nums[i]一定大于nums[i - 2]，因为数组是升序： nums[i - 2] <= nums[i - 1] <= nums[i]，nums[i]最多两个，也就是nums[i] == nums[i + 1]，那么nums[i - 2] < nums[i]。

有数字的个数大于2：[1,1,1,2,2,3,4]

同样除了idx = 0 或 1，如果有数字个数大于2，那么nums[i]会等于nums[i - 2]；至此两个很重要的线索就得到了：1）idx = 0 或 1需要单独考虑；2）有数字个数大于2，则nums[i] == nums[i - 2]；反之，nums[i] > nums[i - 2]。

那么，我们可以初始化序号（idx） = 0，则有下面的步骤：

1. 从头遍历数组，每个数字用n表示；
2. 如果idx < 2，则nums[idx++] = n；
3. idx >= 2，而且nums[idx - 2] < n，则nums[idx++] = n;
4. 最后返回idx；

这里方法的意思是：如果nums[idx - 2] < n，表示这个数最多2个，那么我们重复写入它自己；如果nums[idx - 2] == n，说明有重复的数字，idx需要停留到这个位置，等到下一个nums[idx - 2] < n，表示找到了下一个不重复的数字，则把这个数字写在idx所在的位置，idx后移一位，继续后续检查。而且最后的idx刚好表示数组真实的长度。这部分参考代码：2.2.2 优化版（推荐）。

2.2 实例代码

2.2.1 详细版

class Solution {
    void move(vector<int>& nums, int i, int& count, int& len) {
        int idx = i, diff = count - 2;
        while (idx < len) { nums[idx - diff] = nums[idx]; idx++; }
        len -= diff;
    }

public:
    int removeDuplicates(vector<int>& nums) {
        int len = nums.size(), count = 1;
        for (int i = 0; i < len - 1; i++) {
            if (nums[i] == nums[i + 1]) count++;
            else {
                if (count > 2) { move(nums, i + 1, count, len); i -= count - 2; } // 注意这里i也要回溯，因为数字整体向前移动了
                count = 1;
            }
        }

        if (count > 2) len -= count - 2;
        return len;
    }
};

2.2.2 优化版（推荐）

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int idx  = 0;
        // 写法一
        for (int n : nums)
           if (idx < 2 || n > nums[idx - 2])
               nums[idx++] = n;
               
        // 写法二
        //for (int n = 0; n < nums.size(); n++)
        //    if (idx < 2 || nums[n] > nums[idx - 2])
        //        nums[idx++] = nums[n];
           
        return idx;
    }
};

3. 参考资料

1. 3-6 easy lines, C++, Java, Python, Ruby