【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 <= ) return n;

        int index = ;

        for (int i= ; i<n; i++)

        {

            if ( A[index-]!=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()< ) return nums.size();

            int prev = ;

            for ( int i=; i<nums.size(); ++i )

            {

                if ( nums[prev-]!=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()< ) return nums.size();

            int i=;

            int prev = nums[]!=nums[] ?  : ;

            while (i<nums.size()-)

            {

                if (nums[i]!=nums[i-] && nums[i]!=nums[i+])

                {

                    nums[prev++]=nums[i];

                }

                i++;

            }

            if ( nums[i]!=nums[i-] ) nums[prev++]=nums[i];

            return prev;

        }

};

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

思路很朴素：

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

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

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

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

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

巴特西

【Remove Duplicates from Sorted Array II】cpp

最新文章

热门文章