删除有序数组中的重复元素,并返回数组的新长度

时间:2021-04-07 11:10:29
/**************************************************************
**description:删除有序数组中的重复元素,并返回数组的新长度
** 如:Input:a={2,2,7,15} Output:len=3,a={2,7,15}
***************************************************************/
#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;
//方法一:用index标识新数组的索引,遍历数组元素,当前元素与新数组的最后一个元素不同时,将其加入新数组
//时间复杂度O(n),空间复杂度O(1)
int removeDuplicates_1(vector<int> &arr)
{
if (arr.size() <= 1)
return arr.size();

int index = 0;
for (int i = 1; i < arr.size(); i++)
{
if (arr[i] != arr[index])
arr[++index] = arr[i];
}
return index + 1;
}

//方法二:使用STL库函数,uique()将第一次出现的元素从前往后排,其他重复出现的元素依次排在后面,返回指向重复元素首地址的迭代器
//时间复杂度O(n),空间复杂度O(1)
int removeDuplicates_2(vector<int> &arr)
{
return distance(arr.begin(), unique(arr.begin(), arr.end()));
}

//方法三:利用STL中的upper_bound()查找序列中第一个值大于val的位置
//时间复杂度O(n),空间复杂度O(1)
template<typename InIt, typename OutIt>
OutIt removeDuplicates(InIt first, InIt last, OutIt output)
{
while (first != last)
{
*output++ = *first;
first = upper_bound(first, last, *first);
}
return output;
}
int removeDuplicates_3(vector<int> &arr)
{
return distance(arr.begin(), removeDuplicates(arr.begin(), arr.end(), arr.begin()));
}