给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 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]的赋值操作。