c++中map和set的简单使用

在C++标准库中，map和set是两个非常重要的容器，它们基于红黑树实现，提供了动态集合的功能，并支持高效的查找、插入和删除操作。下面我们将详细探讨map和set的用法和特性。

1. map容器

map是一个关联容器，它包含可以重复的键值对(key-value)，但是每个键(key)在map中必须是唯一的。map中的元素总是按键的升序排序存储。

1.1 map的基本使用

#include <iostream>  
#include <map>  
  
int main() {  
    std::map<std::string, int> myMap;  
  
    // 插入元素  
    myMap["apple"] = 1;  
    myMap["banana"] = 2;  
    myMap["cherry"] = 3;  
  
    // 访问元素  
    std::cout << "Banana count: " << myMap["banana"] << std::endl;  
  
    // 遍历map  
    for (const auto& pair : myMap) {  
        std::cout << pair.first << ": " << pair.second << std::endl;  
    }  
  
    return 0;  
}

1.2 map的常用操作

• find(key): 查找键为key的元素，如果找到，返回指向该元素的迭代器；否则返回end()。
• insert(pair<const Key, T> val): 插入一个键值对。
• erase(iterator pos): 删除迭代器pos指向的元素。
• erase(key_type key): 删除键为key的元素。
• clear(): 清除所有元素。
• size(): 返回元素个数。
• empty(): 判断是否为空。

2. set容器

set是一个关联容器，它包含唯一元素的集合。set中的元素总是按键的升序排序存储。

2.1 set的基本使用

#include <iostream>  
#include <set>  
  
int main() {  
    std::set<int> mySet;  
  
    // 插入元素  
    mySet.insert(1);  
    mySet.insert(2);  
    mySet.insert(3);  
  
    // 遍历set  
    for (const auto& elem : mySet) {  
        std::cout << elem << " ";  
    }  
    std::cout << std::endl;  
  
    return 0;  
}

3. 自定义比较函数或函数对象

默认情况下，map和set使用<操作符进行元素间的比较。但你也可以通过提供自定义的比较函数或函数对象来改变排序规则。这在处理自定义类型或需要非标准排序方式时非常有用。

3.1使用自定义比较函数

bool compareFunc(int a, int b) {
    // 比较逻辑，例如按照绝对值升序排序
    return abs(a) < abs(b);
}

int main() {
    // 使用自定义比较函数来创建map或set对象
    std::map<int, int, bool(*)(int, int)> myMap(compareFunc);
    std::set<int, bool(*)(int, int)> mySet(compareFunc);
    
    // 添加元素到map或set
    myMap[1] = 10;
    myMap[2] = 20;
    
    mySet.insert(5);
    mySet.insert(1);
    mySet.insert(-3);
    
    // 遍历map或set
    for (const auto& pair : myMap) {
        std::cout << pair.first << " " << pair.second << std::endl;
    }
    for (const auto& num : mySet) {
        std::cout << num << " ";
    }
    
    return 0;
}

3.2使用自定义函数对象

struct CompareFunc {
    bool operator()(int a, int b) const {
        // 比较逻辑，例如按照绝对值升序排序
        return abs(a) < abs(b);
    }
};

int main() {
    // 使用自定义函数对象来创建map或set对象
    std::map<int, int, CompareFunc> myMap;
    std::set<int, CompareFunc> mySet;
    
    // 添加元素到map或set（使用与使用默认比较函数相同的方式）
    myMap[1] = 10;
    myMap[2] = 20;
    
    mySet.insert(5);
    mySet.insert(1);
    mySet.insert(-3);
    
    // 遍历map或set（使用与使用默认比较函数相同的方式）
    for (const auto& pair : myMap) {
        std::cout << pair.first << " " << pair.second << std::endl;
    }
    for (const auto& num : mySet) {
        std::cout << num << " ";
    }
    
    return 0;
}
 

4. 性能考虑

由于map和set是基于红黑树实现的，它们的插入、删除和查找操作的时间复杂度都是对数级别的（O(log n)）。这使得它们在处理大量数据时仍然能保持较高的效率。然而，由于它们需要维护树的平衡，因此相比于基于哈希的容器（如unordered_map和unordered_set），map和set在插入和删除操作上可能会稍慢一些。

5. 总结

map和set是C++标准库中非常有用的容器，它们提供了动态集合的功能，并支持高效的查找、插入和删除操作。通过合理使用这两个容器，我们可以更加高效地处理数据，并简化代码的实现。