温馨提示×

C++ map iterator遍历中删除元素

c++
小樊
140
2024-07-30 11:19:10
栏目: 编程语言

在C++中,使用std::map容器的迭代器遍历并删除元素是有一些注意事项的。因为在遍历的过程中删除元素会导致迭代器失效,可能会引起未定义行为。

一种可行的方法是先将需要删除的元素的迭代器存储起来,然后在遍历完成后再统一删除这些元素。示例代码如下:

#include <iostream>
#include <map>

int main() {
    std::map<int, std::string> myMap = {{1, "one"}, {2, "two"}, {3, "three"}};

    std::map<int, std::string>::iterator it;
    std::map<int, std::string>::iterator toDelete;

    // 遍历map并删除需要的元素
    for (it = myMap.begin(); it != myMap.end(); ) {
        if (it->first == 2) {
            toDelete = it;
            ++it;
            myMap.erase(toDelete);
        } else {
            ++it;
        }
    }

    // 输出剩余的元素
    for (it = myMap.begin(); it != myMap.end(); ++it) {
        std::cout << it->first << ": " << it->second << std::endl;
    }

    return 0;
}

在上面的示例中,我们先利用toDelete迭代器存储需要删除的元素的位置,然后在遍历完成后再调用erase方法删除这些元素。这样可以确保不会在遍历过程中出现迭代器失效的情况。

另外,还可以使用std::map容器提供的成员函数erase来删除元素,但需要注意在删除元素后要将迭代器递增,否则会导致迭代器失效。

0