【Remove Duplicates from Sorted Array II】cpp

题目：

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

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

Your function should return length = 5, and A is now [1,1,2,2,3].

 

代码：

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if (n <= 2) return n;
        int index = 2;
        for (int i=2 ; i<n; i++)
        {
            if ( A[index-2]!=A[i] )
            {
                A[index++] = A[i];
            }
        }
        return index;
    }
};

 

Tips:

1. index始终指向下一个要插入元素的位置

2. 判断当前元素与index-2位置元素是否相等，如果不等就可以插入，保证没有连续三个相同的元素

3. 这里用到些数学归纳法的技巧：

　　a. 只要保证第1-第3个元素不是都相同的

　　b. 并且再后面每一步添加元素的时候判断都不是相同的

　则可得结论一直到条件结束，调整后的数组中不会有三个连续重复的元素

========================================

第二次过这道题卡住了一下，复习了第一次写的程式，改出了AC的代码。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
            if ( nums.size()<3 ) return nums.size();
            int prev = 2;
            for ( int i=2; i<nums.size(); ++i )
            {
                if ( nums[prev-2]!=nums[i] )
                {
                    nums[prev++] = nums[i];
                }
            }
            return prev;
    }
};

tips：

这种数学归纳法之类的去重代码，可以总结出一定的规律

第一个指针 prev = 可以保持的重复数量 （这个指针指示当前可插入的位置）

第二个指针 i 从 可以保持重复数量下标开始 一直到数组长度最后

去重条件判断nums[prev-可以保持重复的数量]!=nums[i]

这样就可以获得满足题意的程式。

================================

这里还有一种特殊的case，如果出现重复的就不要了呢？写出了下面的程式

// special case
class SolutionS{
    public:
        static int removeDuplicates(vector<int>& nums)
        {
            if ( nums.size()<2 ) return nums.size();
            int i=1;
            int prev = nums[0]!=nums[1] ? 1 : 0;
            while (i<nums.size()-1)
            {
                if (nums[i]!=nums[i-1] && nums[i]!=nums[i+1])
                {
                    nums[prev++]=nums[i];
                }
                i++;
            }
            if ( nums[i]!=nums[i-1] ) nums[prev++]=nums[i];
            return prev;
        }
};

这个程式是自己写的，自测了一些case。

思路很朴素：

1. 维护一个prev作为待插入的位置

2. 判断一个元素与前后元素是否相等

3. 处理第一个和末尾元素（第一个没有前驱，末尾没有后继，所以要特殊处理）

 ====================================================

还有一种情况，如果数组不是排序的，就用hashtable记录每个数字出现的次数。