LeetCode数组篇80删除有序数组

80. 删除有序数组中的重复项 II

中等

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。

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

我是铁憨憨，复杂度高

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        for(int i=2;i<nums.size();i++){
            if(nums[i]==nums[i-1] && nums[i]==nums[i-2]){
                int begin=i;
                while(i<nums.size() && nums[i]==nums[begin]){
                    i++;
                }
                nums.erase(nums.begin()+begin,nums.begin()+i);
                i=begin;
            }
        }
        return nums.size();
    }
};

 

评论区大神提供思路，我重新写了

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