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

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

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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]的赋值操作。

猜你喜欢

转载自www.cnblogs.com/mini-coconut/p/8976938.html
今日推荐