使用STL从map <t,z>的键填充vector <t> </t,z> </t>

时间:2011-10-05 20:22:53

标签: c++ stl

map<T,Z> m= ...;
vector<T> v;
v.reserve(m.size);
for(map<T,Z>::iterator it=m.begin();it!=m.end();++it)
{
 v.push_back(it->first);
}

是否有使用某些STL功能的更好的1行版本?

编辑:使用c ++ 11!

4 个答案:

答案 0 :(得分:8)

便携式:

struct SelectKey {
  template <typename F, typename S>
  F operator()(const std::pair<const F, S> &x) const { return x.first; }
};

std::transform(m.cbegin(), m.cend(), std::back_inserter(v), SelectKey());

我认为STL的某些实现有一个名为select1st的非标准扩展,它相当于此处显示的SelectKey。正如K-Ballo在评论中指出的那样,还有一个TR1版本。我喜欢明确命名的版本,因为它更容易看到发生了什么。

由于不需要状态,你可以通过使用实际函数而不是函数来减少样板量:

template <typename F, typename S>
F SelectKey()(const std::pair<const F, S> &x) { return x.first; }

std::transform(m.cbegin(), m.cend(), std::back_inserter(v), SelectKey);

如果您可以使用 C ++ 11 ,您可以使用lambda,它将选择代码保持在接近使用位置的位置:

std::transform(m.cbegin(), m.cend(), std::back_inserter(v),
               [](const std::pair<const F, S> &x) { return x.first; });

甚至基于范围的for循环,这可能是最优雅和可读的:

for(const auto &x : m) {
  v.push_back(x.first);
}

答案 1 :(得分:4)

Pre C ++ 11,您可以使用transform和自定义函数struct:

template <class K, class V>
struct key_selector : std::unary_function<const std::pair<K, V>&, const K&>
{
    const K& operator()(const std::pair<K, V>& element) const
    {
        return element.first;
    }
};

transform(m.begin(), m.end(), back_inserter(v), key_selector<T,Z>());

如果您有权访问boost或TR1,则可以将key_selector替换为mem_fn

transform(m.begin(), m.end(), back_inserter(v), mem_fn(&map<T,Z>::value_type::first));

Post-C ++ 11,你可以使用lambdas:

transform(m.begin(), m.end(), back_inserter(v), [](const map<T,Z>::value_type& x) {return x.first;});

答案 2 :(得分:2)

在C ++ 11中,您可以使用lambda表达式:

typedef std::map< std::string, std::string > map_t;
map_t map;
std::vector< std::string > v;

std::for_each(map.begin(), map.end(), [&v](map_t::value_type const& it)
        {
            v.push_back(it.first);
        });

答案 3 :(得分:1)

您可以采取以下措施:

std::transform(m.begin(), m.end(), std::back_inserter(v), FUNCTOR);

FUNCTOR取决于您拥有的STL版本或库和编译器。

C ++ 11(lambda)

std::transform(m.begin(), m.end(), std::back_inserter(v), [](map<T,Z>::const_reference a) { return a.first; });

C ++ 11(std :: get)

std::transform(m.begin(), m.end(), std::back_inserter(v), &std::get<0>);

C ++ SGI STL有一个名为select1st的仿函数,可以使用

std::transform(m.begin(), m.end(), std::back_inserter(v), select1st);

C ++ 03(非C ++ 11)使用类似其他人描述的仿函数对象。