先使用sort函数对数组进行排序,再使用unique函数得到数组去重后的结果。unique函数返回一个迭代器,指向最后一个不重复元素之后的下一个位置。下面代码中的n是,数组去重后,元素的数目。
void get_unique_elements(const std::vector<int> &input, std::vector<int> &output)
{
std::vector<int> temp;
temp.assign(input.begin(), input.end());
for (auto &e : temp) printf("%d ", e);
printf("\n");
std::sort(temp.begin(), temp.end());
for (auto &e : temp) printf("%d ", e);
printf("\n");
int n = std::unique(temp.begin(), temp.end()) - temp.begin();
for (auto &e : temp) printf("%d ", e);
printf("\n");
printf("n = %d\n", n);
for (int i = 0; i < n; ++i) output.push_back(temp[i]);
for (auto &e : output) printf("%d ", e);
printf("\n");
}
测试代码:
std::vector<int> nums = {0, 7, 7, 6, 1, 1, 5, 5, 8, 9};
std::vector<int> nums_unique;
get_unique_elements(nums, nums_unique);
测试结果:
0 7 7 6 1 1 5 5 8 9
0 1 1 5 5 6 7 7 8 9
0 1 5 6 7 8 9 7 8 9
n = 7
0 1 5 6 7 8 9
为实现数组去重功能,unique函数必须搭配sort函数使用,下面测试注释掉get_unique_elements函数中的sort函数时的结果。根据测试结果可知,unique函数只能对相邻的重复元素进行去重。所以去重之前,需要先对数组进行排序。
void get_unique_elements(const std::vector<int> &input, std::vector<int> &output)
{
std::vector<int> temp;
temp.assign(input.begin(), input.end());
for (auto &e : temp) printf("%d ", e);
printf("\n");
//std::sort(temp.begin(), temp.end());
//for (auto &e : temp) printf("%d ", e);
//printf("\n");
int n = std::unique(temp.begin(), temp.end()) - temp.begin();
for (auto &e : temp) printf("%d ", e);
printf("\n");
printf("n = %d\n", n);
for (int i = 0; i < n; ++i) output.push_back(temp[i]);
for (auto &e : output) printf("%d ", e);
printf("\n");
}
测试代码:
std::vector<int> nums = {0, 7, 6, 7, 1, 1, 5, 5, 8, 9};
std::vector<int> nums_unique;
get_unique_elements(nums, nums_unique);
测试结果:
0 7 6 7 1 1 5 5 8 9
0 7 6 7 1 5 8 9 8 9
n = 8
0 7 6 7 1 5 8 9