如何从集合中初始化地图

Question

我想使用std::set（或std::vector）的元素设置std::map 的键。

类似下面的...

std::set<int> keys = { 3,4,6 };
std::map<int,string> results(keys); // syntax error

这可以在不显式迭代集合的情况下完成吗？

Answer 1

你不能。 map 不是 set。它们是根本不同的容器，即使底层结构相似。

---

也就是说，一切皆有可能。如果范围的元素已经排序，std::map 的范围构造函数是线性时间，set 为我们保证。因此，您需要做的就是将变压器应用于每个元素以产生新的范围。最简单的方法是使用 boost::make_transform_iterator 之类的东西（或自己动手）：

template <class K, class F
    class V = decltype(std::declval<F&>()(std::declval<K const&>()))::second_type>
std::map<K, V> as_map(std::set<K> const& s, F&& f) {
    return std::map<K,V>(
        boost::make_transform_iterator(s.begin(), f),
        boost::make_transform_iterator(s.end(), f));
}

std::map<int,string> results =
    as_map(keys, [](int i){
        return std::make_pair(i, std::string{});
    });

如果你总是想要默认初始化，可以简化为：

template <class V, class K>
std::map<K, V> as_map_default(std::set<K> const& s) {
    auto f = [](K const& k) { return std::make_pair(k, V{}); }
    return std::map<K,V>(
        boost::make_transform_iterator(s.begin(), f),
        boost::make_transform_iterator(s.end(), f)); 
}

std::map<int,string> results = as_map_default<string>(keys);

Answer 2

这可以在不显式迭代集合的情况下完成吗？

没有。如果不对其进行迭代，就无法知道集合中的键。您可以编写函数使其显示，就好像存在隐式转换一样，但这些函数最终必须迭代源集合。

一个简单的方法如下：

#include <set>
#include <map>
#include <string>

auto build_map(const std::set<int>& source) -> std::map<int,std::string>
{
  std::map<int,std::string> results;
  for (auto const& i : source) {
    results[i];
  }
  return results;
}

int main()
{
  std::set<int> keys = { 3,4,6 };
  auto results = build_map(keys);
}

当然，如果这样可以提高可读性，我们可以进行模板化：

#include <set>
#include <vector>
#include <unordered_set>
#include <map>
#include <string>
#include <utility>

template<class MappedType, class SourceContainer>
auto build_map(SourceContainer&& source)
{
  using source_type = std::decay_t<SourceContainer>;
  using key_type = typename source_type::value_type;

  std::map<key_type , MappedType> results;
  for (auto const& i : source) {
    results[i];
  }
  return results;
}

int main()
{
  std::set<int> keys = { 3,4,6 };
  auto results = build_map<std::string>(keys);

  // also
  results = build_map<std::string>(std::vector<int>{3, 4, 6});
  results = build_map<std::string>(std::unordered_set<int>{3, 4, 6});
}