c++ - std::map 通过变换替换现有元素

Question

我有一个代码:

std::vector<int> vector = {1, 3, 5, 7, 9};
using my_type = std::pair<int, int>;
std::map<int, boost::optional<my_type>> map;
for (const auto &i : vector) {
    map[i] = boost::none;
}
const my_type val = {1, 5};
std::transform(vector.cbegin(),
               vector.cend(),
               std::inserter(map, map.end()),
               [&val](const int &i) {
                   return std::make_pair(i, boost::optional<my_type>(val));
});

一切正常，但 std::transform 不会通过用现有键替换值来更改映射，所以我有这个:

{ 
    {1, boost::none},
    {3, boost::none},
    {5, boost::none},
    {7, boost::none},
    {9, boost::none},
}

是否可以按照简单的基于范围的方式使其工作？

for (const auto &i : vector) {
    map[i] = boost::optional<my_type>(val));
}

附言我知道 std::inserter 和 std::map::insert 是如何工作的，只是好奇如何更改转换以更改值。

Answer 1

不适用于 inserter，因为 insert 仅在 map 中没有元素时才插入元素。 会有insert_or_assign在 C++17 中。

template<typename Map>
struct map_insert_iterator : public std::iterator<std::output_iterator_tag, void, void, void, void>
{
public:
   map_insert_iterator(Map& m) : map(m)
   {
   }
   map_insert_iterator<Map>& operator = (const typename Map::value_type& value)
   {
      map.insert_or_assign(value.first, value.second);
      return *this;
   }
   map_insert_iterator<Map>& operator * () { return *this; }
   map_insert_iterator<Map>& operator ++ () { return *this; }
   map_insert_iterator<Map>& operator ++ (int) { return *this; }
private:
   Map& map;
};

template<typename Map>
map_insert_iterator<Map> map_inserter(Map& m)
{
   return map_insert_iterator<Map>(m);
}

Live