Leetcode Remove Duplicates from Sorted Array II

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

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

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.



计算去除重复后的array的长度很简单,可以通过一个临时向量来存储去除重复后的array,然后将临时向量赋给array。


代码如下:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        vector<int> temp;
        int count = 0;
        for(int i=0;i<nums.size();i++)
        {
            int tick = 0;
            if(i != nums.size()-1 && nums[i] == nums[i+1])
            {
                while(nums[i] == nums[i+1])
                {
                    tick++;
                    if(tick<3)
                        temp.push_back(nums[i]);
                    i++;
                }
                tick++;
                if(tick<3)
                    temp.push_back(nums[i]);
            }
            else
            {
                temp.push_back(nums[i]);
            }
            
            if(tick >= 2)
                count += 2;
            else
                count ++;
        }
        nums = temp;
        return count;
    }
};


感觉自己写的代码像一坨shit,跟简洁优雅八竿子打不着,需要进行改进。然后就是如何不使用临时变量来实现功能?


代码如下:

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int count = 0;
        int start=0;
        int len = nums.size();
        for(int i=0;i<nums.size();i++)
        {
            int tick = 0;
            if(i != nums.size()-1 && nums[i] == nums[i+1])
            {
                while(nums[i] == nums[i+1])
                {
                    tick++;
                    if(tick<3)
                    {
                        nums[count++] = nums[i];
                    }
                    i++;
                }
                tick++;
                if(tick<3)
                {
                    nums[count++] = nums[i];
                }
            }
            else
            {
                
                nums[count++] = nums[i];
            }
    
        }
        
        while(count < len)
        {
            nums.pop_back();
            count++;
        }
        
        return nums.size();
    }
};


感觉代码太搓了,尤其是第一个if语句简直是一坨shit,和简洁优雅差太多。怎么样才能写出更优雅的代码。参考了下其他人的代码,发现其实只要将array中的array[i]与array[count-2]进行对比就可以了(i代表array遍历的索引,count代表在array上实时生成的新array的索引),如果相等则可以直接去除,否则就可以进行移位。这个方法写出了如此简洁的代码,简直就是TMD的艺术。

具体代码如下:

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





  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值