LeetCode ||Remove Duplicates from Sorted Array II

最新推荐文章于 2022-04-06 16:22:53 发布

xuxumoi

最新推荐文章于 2022-04-06 16:22:53 发布

阅读量303

点赞数

分类专栏： LeetCode 文章标签： Array LeetCode Erase

本文链接：https://blog.csdn.net/xsloop/article/details/48227437

版权

LeetCode 专栏收录该内容

107 篇文章 0 订阅

订阅专栏

方法一：耗时20ms

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        
        if(nums.size() <= 2)
            return nums.size();
            
        int i = 1;
        int pre = nums[0];
        int preCount = 1;
        
        while(i < nums.size())
        {
            if(pre == nums[i])
            {
                preCount++;
                
                if(preCount > 2)//每次出现次数大于2就erase.
                {
                    //需要删除nums[i]
                    nums.erase(nums.begin()+i);
                    continue;
                }
            }
            else
            {
                preCount = 1;
                pre = nums[i];
            }
            i++;
        }
        
        return nums.size();
    }
};

方法二：耗时20ms

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        
        if(nums.size() <= 2)
            return nums.size();
            
        int i = 1;
        int pre = nums[0];
        int preCount = 1;
        
        int small = 1;//small表示移动数组后，指针应该指向的位置。
        
        while(i < nums.size())
        {
            if(pre == nums[i])
            {
                preCount++;
                
                if(small != i)//需要移动i到small 位置上。
                {
                    nums[small] = nums[i];
                }
                if(preCount > 2)
                {
                    small--;
                }
            }
            else
            {
                preCount = 1;
                pre = nums[i];
                
                if(small!=i)
                    nums[small] = nums[i];
            }
            i++;
            small++;
        }
        
        while(small < nums.size())//将后面重复的数字删除。
        {
            nums.pop_back();
        }
        return nums.size();
    }
};