码迷,mamicode.com
首页 > 编程语言 > 详细

c++如何遍历删除map/vector里面的元素

时间:2014-08-14 16:42:08      阅读:220      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   使用   os   io   for   

新技能Get!

问题

对于c++里面的容器, 我们可以使用iterator进行方便的遍历. 但是当我们通过iterator对vector/map等进行修改时, 我们就要小心了, 因为操作往往会导致iterator失效, 之后的行为都变得不可预知. 比如:

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    vector<int> a = {12, 23, 34, 45, 56, 67, 78, 89};
    
    for (auto iter = a.begin(); iter != a.end(); ++iter) {
        if (*iter > 30) {
            a.erase(iter);
        }
    }
    
    for (const auto &element : a) {
        cout<<element<<endl;
    }
    
    return 0;
}

输出:

12
23
45
67
89

cplusplus的reference里对 std::vector::erase 的描述是:

Iterators, pointers and references pointing to position (or first) and beyond are invalidated, with all iterators, pointers and references to elements before position (or first) are guaranteed to keep referring to the same elements they were referring to before the call.

只有删除元素前面的iterator还保持有效, 之后的遍历行为不可预知.

解决方案

对于vector, erase会返回下一个iterator, 因此我们可以使用如下的方法:

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    vector<int> a = {12, 23, 34, 45, 56, 67, 78, 89};
    
    auto iter = a.begin();
    while (iter != a.end()) {
        if (*iter > 30) {
            iter = a.erase(iter);
        }
        else {
            ++iter;
        }
    }
    
    for (const auto &element : a) {
        cout<<element<<endl;
    }
    
    return 0;
}


输出:

12
23

  

对于map, 删除iterator只会影响当前的iterator, 因此使用for循环就够了, 比如:

#include <iostream>
#include <map>

using namespace std;

int main()
{
    map<int, int> a = {{1, 12}, {2, 23}, {3, 34}, {4, 45}, {5, 56}, {6, 67}};
    
    for (auto iter = a.begin(); iter != a.end(); ++iter) {
        if (iter->second > 30) {
            a.erase(iter);
        }
    }
    
    for (const auto &element : a) {
        cout<<element.first<<" : "<<element.second<<endl;
    }
    
    return 0;
}

输出:

1 : 12
2 : 23

  但是更推荐的做法是在erase前让iterator指向下一个元素

#include <iostream>
#include <map>

using namespace std;

int main()
{
    map<int, int> a = {{1, 12}, {2, 23}, {3, 34}, {4, 45}, {5, 56}, {6, 67}};
    
    auto iter = a.begin();
    while (iter != a.end()) {
        if (iter->second > 30) {
            a.erase(iter++);
        }
        else {
            ++iter;
        }
    }
    
    for (const auto &element : a) {
        cout<<element.first<<" : "<<element.second<<endl;
    }
    
    return 0;
}

输出:

1 : 12
2 : 23

  

参考资料

http://stackoverflow.com/questions/4645705/vector-erase-iterator

http://stackoverflow.com/questions/4600567/how-can-i-delete-elements-of-a-stdmap-with-an-iterator

c++如何遍历删除map/vector里面的元素,布布扣,bubuko.com

c++如何遍历删除map/vector里面的元素

标签:style   blog   http   color   使用   os   io   for   

原文地址:http://www.cnblogs.com/dabaopku/p/3912662.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!