unique函数是STL中用于删除容器中重复元素的函数,前提是容器中的内容已经有序排列,(LeetCode 26题实现同样的功能),STL源码是非常好的阅读材料,里面许多算法值得阅读和推敲
LeetCode 26题
/* 给定一个排序数组,你需要在 原地 删除重复出现的元素,
使得每个元素只出现一次,返回移除后数组的新长度。*/
/* 不要使用额外的数组空间,你必须在 原地 修改输入数组
并在使用 O(1) 额外空间的条件下完成。*/
int removeDuplicates(vector<int>& nums) {
if(nums.begin() == nums.end()) return 0;
auto itbegin = nums.begin();
auto result = itbegin;
while(++itbegin != nums.end())
if(*result != *itbegin)
*++result = *itbegin;
return result - nums.begin() + 1;
}
// 更简单的写法
int removeDuplicates(vector<int>& nums) {
if(nums.size() < 2) return nums.size();
int j = 0;
for (int i = 1; i < nums.size(); i++)
if (nums[j] != nums[i])
nums[++j] = nums[i];
return ++j;
}
C++ STL unique源码
template<typename iterator>
iterator unique(iterator first, iterator last)
{
if(first == last) return last;
iterator result = first;
while(++first != last)
if(*result != *first)
*++result = *first;
return ++result;
}
分析这段代码,我们可以知道,unique函数的去重过程实际上就是不停的把后面不重复的元素移到前面来,也可以说是用不重复的元素占领重复元素的位置。