将选定字段从结构上的无序集合存储到向量

时间:2019-04-09 09:55:48

标签: c++ stl stdvector unordered-set

我有一个unordered_set存储以下结构

struct match_t{
  size_t score;
  size_t ci;  
};

typedef std::unordered_set<match_t> uniq_t;

现在,我想将uniq_t myset;的元素存储到向量中,但是这样做时,我只想复制分数而不是整个struct。我已经看到了使用assignback_inserter分配元素的解决方案。我想知道如何从结构中选择所需的字段。为此,我在assignback_inserter中看不到任何参数。

我应该尝试为向量重写push_back方法,还是有其他方法可以做到这一点?

编辑1 通过使用这些方法中的任何一种,而不是遍历集合并分配所需的值,是否可以获得任何性能改进?

1 个答案:

答案 0 :(得分:5)

简单的for循环没什么错:

std::unordered_set<match_t> myset;
std::vector<std::size_t> myvec;

myvec.reserve(myset.size()); // allocate memory only once

for (const auto& entry : myset)
    myvec.push_back(entry.score);

或者,您可以将std::transform与自定义lambda结合使用:

#include <algorithm>

std::tranform(myset.cbegin(), myset.cend(), std::back_inserter(myvec),
    [](const auto& entry){ return entry.score; });

另一种方法是使用范围库,例如使用range-v3

#include <range/v3/view/transform.hpp>

std::vector<std::size_t> myvec = myset | ranges::view::transform(&match_t::score);

在性能方面,您不能对所有match_t对象进行线性传递。相反,重要的调整是最大程度地减少分配数量。由于结果std::vector的大小是先验已知的,因此如上所述调用std::vector::reserve可以确保不会发生不必要的分配。