Получение списка ключей и значений из unordered_map

Каков наиболее эффективный способ получения списков (как vector) ключей и значений из unordered_map?

Для конкретности предположим, что рассматриваемая карта a unordered_map<string, double>. Я хотел бы получить ключи как vector<string>, а значения - как vector<double>.

unordered_map<string, double> um;

vector<string> vs = um.enum_keys();
vector<double> vd = um.enum_values();

Я могу просто перебрать карту и собрать результат, но есть ли еще эффективный метод? Было бы неплохо иметь метод, который также работает для обычной карты, так как я могу переключиться на это.

Ответ 1

Хорошо, вот вы идете:

std::vector<Key> keys;
keys.reserve(map.size());
std::vector<Val> vals;
vals.reserve(map.size());

for(auto kv : map) {
    keys.push_back(kv.first);
    vals.push_back(kv.second);  
}

Эффективность, вероятно, может быть улучшена, но она есть. Вы работаете на двух контейнерах, хотя на самом деле не существует какой-либо магии STL, которая может скрыть этот факт.

Как сказал Луис, это будет работать для любых контейнеров STL map или set.

Ответ 2

Используя С++ - 14, вы также можете сделать следующее (отредактированное для полного источника):

#include <algorithm>
#include <iostream>
#include <string>
#include <unordered_map>
#include <vector>

using namespace std;

typedef string Key;
typedef int Value;

auto key_selector = [](auto pair){return pair.first;};
auto value_selector = [](auto pair){return pair.second;};

int main(int argc, char** argv) {
  // Create a test map
  unordered_map<Key, Value> map;
  map["Eight"] = 8;
  map["Ten"] = 10;
  map["Eleven"] = 11;

  // Vectors to hold keys and values
  vector<Key> keys(map.size());
  vector<Value> values(map.size());

  // This is the crucial bit: Transform map to list of keys (or values)
  transform(map.begin(), map.end(), keys.begin(), key_selector);
  transform(map.begin(), map.end(), values.begin(), value_selector);

  // Make sure this worked: Print out vectors
  for (Key key : keys) cout << "Key: " << key << endl;
  for (Value value : values) cout << "Value: " << value << endl;

  return 0;
}

Я скомпилировал это с помощью следующей команды:

g++ keyval.cpp -std=c++14 -o keyval

Тестирование распечатало ключи и значения, как ожидалось.

Ответ 3

В STL нет встроенного метода для получения всех ключей или значений с карты.

Невозможно перебрать неупорядоченную карту или обычную карту, лучший способ - перебрать ее и собрать ключ или значение для вектора.

Вы можете написать функцию шаблона для итерации любого вида карты.

Ответ 4

Вступление поздно, но подумал, что это может кому-то помочь.
Две функции шаблона используют key_type и mapped_type.

namespace mapExt
{
    template<typename myMap>
    std::vector<typename myMap::key_type> Keys(const myMap& m)
    {
        std::vector<typename myMap::key_type> r;
        r.reserve(m.size());
        for (const auto&kvp : m)
        {
            r.push_back(kvp.first);
        }
        return r;
    }

    template<typename myMap>
    std::vector<typename myMap::mapped_type> Values(const myMap& m)
    {
        std::vector<typename myMap::mapped_type> r;
        r.reserve(m.size());
        for (const auto&kvp : m)
        {
            r.push_back(kvp.second);
        }
        return r;
    }
}

Использование:

std::map<long, char> mO;
std::unordered_map<long, char> mU;
// set up the maps
std::vector<long> kO = mapExt::Keys(mO);
std::vector<long> kU = mapExt::Keys(mU);
std::vector<char> vO = mapExt::Values(mO);
std::vector<char> vU = mapExt::Values(mU);