使用erase(begin())从unordered_set中“删除一个元素”在gcc中很慢

时间:2012-04-17 16:46:21

标签: c++ erase unordered-set

我想从unordered_set中删除“删除一个元素”的功能。

然而,当使用erase(begin())实现它时,它变得非常慢。 (这是用g ++ - 4.5.3;也许begin()必须遍历大量的空哈希桶?)

请参阅下面的示例代码,并提供令人惊讶的时间。

是否有其他方法可以实现“删除一个元素”,从而提高效率? (我确实想允许其他干预集合操作,这会使迭代器无效。)

#include <unordered_set>
#include <iostream>
#include <chrono>
using namespace std;

struct Timer {
    Timer(string s) : _s(s), _start(Clock::now()) { }
    ~Timer() {
        auto t=chrono::duration_cast<chrono::milliseconds>(Clock::now()-_start).count();
        cerr << "Timer(" << _s << ") = " << t << "ms\n";
    }
 private:
    typedef chrono::high_resolution_clock Clock;
    string _s;
    Clock::time_point _start;
};

int main()
{
    unordered_set<int> s;
    const int num=200000;
    { Timer t("insert"); for (int i=0;i<num;i++) { s.insert(i); } }
    { Timer t("remove half"); for (int i=0;i<num/2;i++) { s.erase(s.begin()); } }
    long long s1=0, s2=0;
    { Timer t("access begin()"); for (int i=0;i<num/2;i++) { s1+=*s.begin(); } }
    { Timer t("access all"); for (auto it=s.begin();it!=s.end();++it) { s2+=*it; } }
    cerr << s1 << " " << s2 << "\n";
    return 0;
}

// Timer(insert) = 36ms
// Timer(remove half) = 3039ms
// Timer(access begin()) = 5958ms
// Timer(access all) = 1ms

1 个答案:

答案 0 :(得分:6)

这个版本的GNU库看起来像一个问题,在更新的版本中得到修复。以下是我的测试结果,使用了我碰巧安装的两个版本:

mikes@seymour-desktop:~$ g++-4.4.5 -std=c++0x -O3 test.cpp
mikes@seymour-desktop:~$ ./a.out 
Timer(insert) = 15ms
Timer(remove half) = 3815ms
Timer(access begin()) = 7722ms
Timer(access all) = 0ms
10000000000 14999950000

mikes@seymour-desktop:~$ g++-4.6.1 -std=c++0x -O3 test.cpp
mikes@seymour-desktop:~$ ./a.out 
Timer(insert) = 16ms
Timer(remove half) = 2ms
Timer(access begin()) = 0ms
Timer(access all) = 1ms
10000000000 14999950000

通过使用boost::unordered_set,我也获得了类似的快速结果,因此如果您无法更新编译器,那么这是一个选项。