std::map

std::map 是 C++ 标准模板库（STL）中的一种关联容器，用于存储键值对（key-value pairs），并且键是有序的。它底层通常使用 红黑树（Red-Black Tree） 来实现。红黑树是一种自平衡的二叉搜索树，保证了在最坏情况下基本操作的时间复杂度为 O(log n)。

1. 底层实现：红黑树

红黑树是一种平衡二叉搜索树，具有以下性质：

• 每个节点要么是红色，要么是黑色。
• 根节点是黑色。
• 每个叶子节点（即空节点，NIL 节点）是黑色。
• 如果一个节点是红色的，则它的两个子节点都是黑色的（即红节点不能有红子节点，也不能连续两个红节点）。
• 从任一节点到其每个叶子的所有路径都包含相同数目的黑色节点。

红黑树的这些性质确保了树的高度近似平衡，从而保证了插入、删除和查找操作的时间复杂度为 O(log n)。

2. std::map 的操作

由于 std::map 使用红黑树实现，这些操作的时间复杂度通常为 O(log n)。

插入操作

• insert(const value_type& value)：将一个键值对插入到 map 中。如果键已经存在，则插入失败。
• operator[]：可以通过 map[key] = value 的方式插入或更新一个元素。如果键不存在，则会创建一个默认值。

插入时，红黑树会找到适当的位置插入新节点，并可能通过旋转和重新着色来维护红黑树的平衡性。

查找操作

• find(const key_type& key)：查找指定键的元素，返回指向该元素的迭代器。如果未找到，则返回 end()。
• count(const key_type& key)：判断键是否存在于 map 中，返回 0 或 1。

查找操作利用了红黑树的二叉搜索树性质，从根节点开始，根据键值的大小决定是向左子树还是右子树移动，直到找到目标键或达到叶子节点。

删除操作

• erase(const key_type& key)：删除指定键的元素。
• erase(iterator pos)：删除指定位置的元素。

删除操作同样涉及维护红黑树的性质。删除节点后，可能需要通过旋转和重新着色来重新平衡树。

迭代器

• begin() 和 end()：返回指向 map 首元素和尾后位置的迭代器。
• rbegin() 和 rend()：返回指向 map 最后一个元素和第一个元素前位置的反向迭代器。

由于 std::map 中的键是有序的，因此迭代器遍历时会按键的顺序进行。

3. 具体操作示例

以下是一个使用 std::map 的简单示例：

#include <iostream>
#include <map>

int main() {
    std::map<int, std::string> myMap;

    // 插入元素
    myMap.insert({1, "one"});
    myMap.insert({3, "three"});
    myMap.insert({2, "two"});

    // 使用 operator[] 插入或修改元素
    myMap[4] = "four";

    // 查找元素
    auto it = myMap.find(3);
    if (it != myMap.end()) {
        std::cout << "Found: " << it->first << " -> " << it->second << std::endl;
    }

    // 遍历 map
    for (const auto& [key, value] : myMap) {
        std::cout << key << ": " << value << std::endl;
    }

    // 删除元素
    myMap.erase(2);

    std::cout << "After erase:" << std::endl;
    for (const auto& [key, value] : myMap) {
        std::cout << key << ": " << value << std::endl;
    }

    return 0;
}

4. 总结

• 有序性：std::map 保证所有键值对按键有序存储，适合需要保持有序的数据集合。
• 时间复杂度：插入、查找和删除操作的时间复杂度为 O(log n)，因为这些操作依赖红黑树的高度。
• 迭代器稳定性：插入和删除操作不会使其他迭代器失效，除非删除的就是该迭代器指向的元素。