Fork me on GitHub

Leetcode80. Remove Duplicates from Sorted Array II删除排序数组中的重复项2

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

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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 。 你不需要考虑数组中超出新长度后面的元素。

错误:

class Solution {
public:
    int removeDuplicates(vector<int>& nums)
    {
        int len = nums.size();
        if(len <= 2)
            return len;
        int pos = 2;
        for(int i = 2; i < len; i++)
        {
            if(nums[i] == nums[i - 1] && nums[i] == nums[i - 2])
            {
                continue;
            }
            else
            {
                nums[pos++] = nums[i];
            }
        }
        return pos;
    }
};

如果是111223这种情况的话,如果按照上面算法的情况,把把2赋值给前面重复的位置就变成了112223,这样到最后结果就为1123,判定了2的重复次数大于2.但是2的次数并没有大于2。所以要在上面算法的情况下新加一个判定,再设置个变量记录最后一个重复的被覆盖的元素。比如11223中的1。

class Solution {
public:
    int removeDuplicates(vector<int>& nums)
    {
        int len = nums.size();
        if(len <= 2)
            return len;
        int pos = 2;
        int last = 0;
        bool isLast = false;
        for(int i = 2; i < len; i++)
        {
            if(nums[i] == nums[i - 1] && nums[i] == nums[i - 2])
            {
                if(isLast && i - pos == 1 && nums[i] != last)
                {
                    last = nums[pos];
                    nums[pos++] = nums[i];
                    continue;
                }
            }
            else
            {
                isLast = true;
                last = nums[pos];
                nums[pos++] = nums[i];
            }
        }
        return pos;
    }
};

 

posted @ 2018-11-23 23:42  lMonster81  阅读(79)  评论(0编辑  收藏  举报
/*评论*/ /*top按钮*/

/* 网易云控件 */