LeetCode 80. 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.

题意：给你一个有序数组nums，对于重复的数字保留两个，返回元素的个数。

思路：当然这个题用之前的方法也是可行的，用erase()函数，当有连续相等的三个元素就擦除最后一个，

注意：传入的数组为空或者元素个数小于2的情况

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.size()<2) return nums.size();
        for (auto n = nums.begin(); n != nums.end()-2;)
        {
        	if (*n == *(n + 2))
        	{
        		n = nums.erase(n);
        	}
        	else
        	{
        		n++;
        	}
        
        }
        
        return nums.size();
    }
};

从着几道题目可以看出这样的效率十分低下：

考虑之前的优化实现方法加一个索引count，思路想不出。

参考别人的代码，惊为天人，代码巧妙，十分佩服。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
         int i = 0;
         for (int n : nums)
            if (i < 2 || n > nums[i-2])
                nums[i++] = n;
        return i;
    }
};

有效代码3行。

1.遍历整个数组，当索引值i小于2时(或元素个数小于3)，不动原数组元素，同时索引值i++，有效范围[0...i)

2.当索引值大于2时，就和它前边距离为2的做比较，不相等索引值移位