在C++编程中,我们经常需要处理包含重复元素的vector。传统的方法可能需要嵌套循环或复杂的逻辑,既繁琐又低效。今天,我们将探讨如何利用C++ STL(标准模板库)的强大功能,用简洁优雅的代码高效地删除vector中的重复元素。
STL提供了一系列强大的算法和容器,其中std::sort
和std::unique
这两个算法配合使用,可以轻松实现vector的去重。下面是详细的步骤和代码实现:
#include <iostream>
#include <vector>
#include <algorithm>
void printVector(const std::vector<int>& vec) {
for (const auto& elem : vec) {
std::cout << elem << " ";
}
std::cout << std::endl;
}
int main() {
// 1. 创建一个包含重复元素的vector
std::vector<int> numbers = {3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5};
std::cout << "原始vector: ";
printVector(numbers);
// 2. 对vector进行排序
std::sort(numbers.begin(), numbers.end());
std::cout << "排序后: ";
printVector(numbers);
// 3. 使用std::unique重新排列元素,将重复元素移到末尾
auto last = std::unique(numbers.begin(), numbers.end());
std::cout << "unique后: ";
printVector(numbers);
// 4. 删除重复元素
numbers.erase(last, numbers.end());
// 5. 输出结果
std::cout << "去重后的vector: ";
printVector(numbers);
return 0;
}
让我们逐步解析这个过程:
-
首先,我们创建一个包含重复元素的vector。
-
使用
std::sort
对vector进行排序。这是必要的,因为std::unique
只能识别相邻的重复元素。 -
调用
std::unique
。这个函数不会真正删除元素,而是将不重复的元素移到vector的前部,并返回一个迭代器,指向最后一个不重复元素之后的位置。 -
使用
vector::erase
函数删除从last
迭代器到vector末尾的所有元素,这些就是重复的元素。 -
最后,我们得到了一个没有重复元素的vector。
这种方法的优点:
- 简洁:只需几行代码就能完成复杂的去重操作。
- 高效:时间复杂度为O(nlogn),主要来自排序步骤。
- 通用:可以处理任何支持比较操作的元素类型。
注意事项:
- 这种方法会改变原vector中元素的顺序。
- 如果需要保持原有顺序,可以考虑使用
std::set
或std::unordered_set
。
结论:
通过巧妙运用C++ STL,我们可以用简洁的代码高效地解决vector去重这类常见问题。这不仅提高了代码的可读性,也大大提升了程序的执行效率。在日常编程中,善用STL可以让我们的代码更加优雅和高效。