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

Posted 2020-11-02 Mini_Coconut

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

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