码迷,mamicode.com
首页 > 编程语言 > 详细

Leetcode(8)-删除排序数组中的重复项

时间:2018-05-01 20:28:33      阅读:256      评论:0      收藏:0      [点我收藏+]

标签:一个   tps   ret   ==   ike   vector   href   如何   ras   

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

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

我们利用STL中vector现有的接口完美实现了该功能,代码如下

int removeDuplicates(vector<int>& nums) 
{
    nums.erase(unique(nums.begin(),nums.end()),nums.end());
    return nums.size();
}

其中unique函数可以将容器中相邻的重复元素,只留一个,其余的放入vector的后面,返回值是容器去重之后的尾地址。我们还要借用erase函数将后面的重复元素删除掉,最后再返回容器的大小。

上述方法简洁而且完美,但是毕竟是STL中的内部接口,我们还是要思考下,在不使用额外空间的前提下,内部如何实现该功能

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

这样就平凡多了,但看起来也更踏实点。核心在遍历的同时计算重复元素的个数,还有nums[i-count]=nums[i]的赋值操作。

Leetcode(8)-删除排序数组中的重复项

标签:一个   tps   ret   ==   ike   vector   href   如何   ras   

原文地址:https://www.cnblogs.com/mini-coconut/p/8976938.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!