Как извлечь все ключи (или значения) из std :: map и поместить их в вектор?


246

Это один из возможных способов выхода:

struct RetrieveKey
{
    template <typename T>
    typename T::first_type operator()(T keyValuePair) const
    {
        return keyValuePair.first;
    }
};

map<int, int> m;
vector<int> keys;

// Retrieve all keys
transform(m.begin(), m.end(), back_inserter(keys), RetrieveKey());

// Dump all keys
copy(keys.begin(), keys.end(), ostream_iterator<int>(cout, "\n"));

Конечно, мы также можем извлечь все значения из карты, определив другой функтор RetrieveValues .

Есть ли другой способ достичь этого легко? (Мне всегда интересно, почему std :: map не включает функцию-член для нас.)


10
Ваше решение самое лучшее ...
linello

4
Единственное, что я хотел бы добавить, это keys.reserve(m.size());.
Галик

Ответы:


176

Хотя ваше решение должно работать, его может быть сложно прочитать в зависимости от уровня навыков ваших коллег-программистов. Кроме того, он перемещает функциональность от сайта вызова. Что может сделать обслуживание немного сложнее.

Я не уверен, что ваша цель - вставить ключи в вектор или напечатать их для cout, поэтому я делаю и то, и другое. Вы можете попробовать что-то вроде этого:

map<int, int> m;
vector<int> v;
for(map<int,int>::iterator it = m.begin(); it != m.end(); ++it) {
  v.push_back(it->first);
  cout << it->first << "\n";
}

Или даже проще, если вы используете Boost:

map<int,int> m;
pair<int,int> me; // what a map<int, int> is made of
vector<int> v;
BOOST_FOREACH(me, m) {
  v.push_back(me.first);
  cout << me.first << "\n";
}

Лично мне нравится версия BOOST_FOREACH, потому что она меньше печатает и очень четко рассказывает о том, что она делает.


1
Пойди разберись, что бы я вернулся сюда после поиска в Google. Ваш ответ я предпочитаю :)
mpen

4
@ Jere - Ты действительно работал с BOOST_FOREACH? Код, который вы предлагаете здесь, абсолютно неверен
Мануэль

2
@Jamie - это другой способ, но в документах о повышении указывается указание переменной и ее типа перед BOOST_FOREACH, если тип содержит запятую. Они также показывают typedefing это. Итак, я запутался, что не так с моим кодом?
Jere.Jones

17
Любопытно, не имеет ли смысла задавать вектор, чтобы предотвратить распределение размера?
Алан

2
Не забудьте сделать, v.reserve(m.size())чтобы избежать изменения размера вектора во время передачи.
Брайан Уайт

157
//c++0x too
std::map<int,int> mapints;
std::vector<int> vints;
vints.reserve(mapints.size());
for(auto const& imap: mapints)
    vints.push_back(imap.first);

4
Ницца. Забудь о it = ...begin(); it != ...end. Самым хорошим будет, конечно, std :: map с методом keys (), возвращающим этот вектор ...
masterxilo

2
@BenHymers: мне кажется, что этот ответ был дан по адресу answered Mar 13 '12 at 22:33, то есть через несколько месяцев после того, как C ++ 11 стал C ++.
Себастьян Мах

37
@BenHymers, но он полезен всем, кто сейчас читает вопрос, в чем суть SO - не только помощь спрашивающему, но и всем остальным.
Лучиан Григоре

9
for (auto & imap) является более точным, потому что нет операции копирования.
HelloWorld

2
@StudentT, еще лучше for(auto const & imap : mapints).
cp.engr

61

Для этого есть адаптер диапазона усиления :

vector<int> keys;
// Retrieve all keys
boost::copy(m | boost::adaptors::map_keys, std::back_inserter(keys));

Есть аналогичный адаптер диапазона map_values ​​для извлечения значений.


1
К сожалению, похоже, что boost::adaptorsне доступен до Boost 1.43. Текущий стабильный выпуск Debian (Squeeze) предлагает только Boost 1.42
Mickaël Le Baillif

2
Какая жалость. Boost 1.42 был выпущен в феврале 2010 года, за 2,5 года до Squeeze.
Аластер

На этом этапе не должны ли Squeeze Updates и / или репозиторий Backports предлагать Boost 1.44?
Луис Мачука

какой заголовок Boost определен в?
Джеймс Вежба

1
См. Связанный документ, это определено вboost/range/adaptor/map.hpp
Аластер

47

C ++ 0x дал нам еще одно отличное решение:

std::vector<int> keys;

std::transform(
    m_Inputs.begin(),
    m_Inputs.end(),
    std::back_inserter(keys),
    [](const std::map<int,int>::value_type &pair){return pair.first;});

22
На мой взгляд, в этом нет ничего превосходного. ключи std :: vector <int>; keys.reserve (m_Inputs.size ()); for (auto keyValue: m_Inputs) {keys.push_back (keyValue.first); } Гораздо лучше, чем загадочное преобразование. Даже с точки зрения производительности. Этот лучше.
Джаганнатха

5
Вы также можете зарезервировать размер клавиш здесь, если вы хотите сравнимую производительность. используйте преобразование, если вы хотите избежать цикла for.
DanDan

4
просто хочу добавить - можно использовать [] (const auto & pair)
ivan.ukr

@ ivan.ukr какой компилятор вы используете? Этот синтаксис здесь недопустим: 'const auto &': параметр не может иметь тип, который содержит 'auto'
Гоб

4
Автоматический параметр @ ivan.ukr в лямбде - это c ++ 14
roalz

16

@ DanDan ответ, используя C ++ 11:

using namespace std;
vector<int> keys;

transform(begin(map_in), end(map_in), back_inserter(keys), 
            [](decltype(map_in)::value_type const& pair) {
    return pair.first;
}); 

и используя C ++ 14 (как отмечено @ ivan.ukr), мы можем заменить decltype(map_in)::value_typeна auto.


5
Вы могли бы добавить keys.reserve(map_in.size());для эффективности.
Галик

Я считаю, что метод преобразования на самом деле занимает больше кода, чем цикл for.
user1633272

const можно поставить за типом! Я почти забыл это.
Чжан


10

Ваше решение хорошо, но вы можете использовать итератор для этого:

std::map<int, int> m;
m.insert(std::pair<int, int>(3, 4));
m.insert(std::pair<int, int>(5, 6));
for(std::map<int, int>::const_iterator it = m.begin(); it != m.end(); it++)
{
    int key = it->first;
    int value = it->second;
    //Do something
}

10

Основано на решении @ rusty-parks, но в c ++ 17:

std :: map <int, int> items;
std :: vector <int> itemKeys;

для (const auto & [ключ, игнорируется]: элементы)
{
    itemKeys.push_back (ключ);
}

Я не думаю, что таким образом std::ignoreможно использовать структурированные привязки. Я получаю ошибку компиляции. Достаточно просто использовать обычную переменную, например, ignoredкоторая просто не используется.
ДБ

1
@jb Спасибо. Действительно, std::ignoreпредназначен для использования, std::tieно не со структурными привязками. Я обновил свой код
Мадияр

9

Я думаю, что BOOST_FOREACH, представленный выше, хорош и чист, однако есть и другой вариант, использующий BOOST.

#include <boost/lambda/lambda.hpp>
#include <boost/lambda/bind.hpp>

std::map<int, int> m;
std::vector<int> keys;

using namespace boost::lambda;

transform(      m.begin(), 
                m.end(), 
                back_inserter(keys), 
                bind( &std::map<int,int>::value_type::first, _1 ) 
          );

copy( keys.begin(), keys.end(), std::ostream_iterator<int>(std::cout, "\n") );

Лично я не думаю, что этот подход такой же чистый, как подход BOOST_FOREACH в этом случае, но boost :: lambda может быть действительно чистым в других случаях.


7

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


7

Бит с ++ 11 взять:

std::map<uint32_t, uint32_t> items;
std::vector<uint32_t> itemKeys;
for (auto & kvp : items)
{
    itemKeys.emplace_back(kvp.first);
    std::cout << kvp.first << std::endl;
}

5

Вы можете использовать универсальный boost :: transform_iterator. Transform_iterator позволяет вам преобразовывать повторяющиеся значения, например, в нашем случае, когда вы хотите иметь дело только с ключами, а не со значениями. См. Http://www.boost.org/doc/libs/1_36_0/libs/iterator/doc/transform_iterator.html#example


5

Вот хороший шаблон функции с использованием магии C ++ 11, работающей как для std :: map, так и для std :: unordered_map:

template<template <typename...> class MAP, class KEY, class VALUE>
std::vector<KEY>
keys(const MAP<KEY, VALUE>& map)
{
    std::vector<KEY> result;
    result.reserve(map.size());
    for(const auto& it : map){
        result.emplace_back(it.first);
    }
    return result;
}

Проверьте это здесь: http://ideone.com/lYBzpL


4

Лучшее не-sgi, не поддерживающее STL решение - это расширить map :: iterator следующим образом:

template<class map_type>
class key_iterator : public map_type::iterator
{
public:
    typedef typename map_type::iterator map_iterator;
    typedef typename map_iterator::value_type::first_type key_type;

    key_iterator(const map_iterator& other) : map_type::iterator(other) {} ;

    key_type& operator *()
    {
        return map_type::iterator::operator*().first;
    }
};

// helpers to create iterators easier:
template<class map_type>
key_iterator<map_type> key_begin(map_type& m)
{
    return key_iterator<map_type>(m.begin());
}
template<class map_type>
key_iterator<map_type> key_end(map_type& m)
{
    return key_iterator<map_type>(m.end());
}

а затем использовать их так:

        map<string,int> test;
        test["one"] = 1;
        test["two"] = 2;

        vector<string> keys;

//      // method one
//      key_iterator<map<string,int> > kb(test.begin());
//      key_iterator<map<string,int> > ke(test.end());
//      keys.insert(keys.begin(), kb, ke);

//      // method two
//      keys.insert(keys.begin(),
//           key_iterator<map<string,int> >(test.begin()),
//           key_iterator<map<string,int> >(test.end()));

        // method three (with helpers)
        keys.insert(keys.begin(), key_begin(test), key_end(test));

        string one = keys[0];

1
Я оставлю читателю также возможность создавать const_iterator и обратные итераторы, если / когда это необходимо.
Мариус

-1

На примере атомарной карты

#include <iostream>
#include <map>
#include <vector> 
#include <atomic>

using namespace std;

typedef std::atomic<std::uint32_t> atomic_uint32_t;
typedef std::map<int, atomic_uint32_t> atomic_map_t;

int main()
{
    atomic_map_t m;

    m[4] = 456;
    m[2] = 45678;

    vector<int> v;
    for(map<int,atomic_uint32_t>::iterator it = m.begin(); it != m.end(); ++it) {
      v.push_back(it->second);
      cout << it->first << " "<<it->second<<"\n";
    }

    return 0;
}

-2

Немного похоже на один из примеров, упрощенный с std::mapточки зрения использования.

template<class KEY, class VALUE>
std::vector<KEY> getKeys(const std::map<KEY, VALUE>& map)
{
    std::vector<KEY> keys(map.size());
    for (const auto& it : map)
        keys.push_back(it.first);
    return keys;
}

Используйте как это:

auto keys = getKeys(yourMap);

2
Эй, я знаю, что этот ответ старый, но он тоже неправильный. Инициализация с размером map.size()означает двойной возврат размера вектора. Пожалуйста , исправьте , чтобы спасти кого - то еще головная боль :(
THC

-3

(Мне всегда интересно, почему std :: map не включает функцию-член для нас.)

Потому что он не может сделать это лучше, чем ты можешь. Если реализация метода не превосходит реализацию свободной функции, то в общем случае вы не должны писать метод; Вы должны написать бесплатную функцию.

Также не сразу понятно, почему это так или иначе полезно.


8
Существуют и другие причины, помимо эффективности, предоставляемой библиотекой для предоставления метода, такие как функциональность «батареи включены» и согласованный, инкапсулированный API. Хотя по общему признанию ни один из этих терминов не описывает STL особенно хорошо :) Re. непонятно, зачем это полезно - правда? Я думаю, что совершенно очевидно, почему перечисление доступных ключей полезно для карты / dict: это зависит от того, для чего вы ее используете.
Эндибакли

4
По этой empty()причине мы не должны иметь, потому что это может быть реализовано как size() == 0.
сентября

1
Что сказал @ gd1. Хотя в классе не должно быть большого количества функциональной избыточности, настаивать на том, что абсолютный ноль не является хорошей идеей IMO - по крайней мере, до тех пор, пока C ++ не позволит нам «благословлять» свободные функции в методах.
einpoklum

1
В более старых версиях C ++ были контейнеры, для которых empty () и size () могли разумно иметь разные гарантии производительности, и я думаю, что спецификация была достаточно свободной, чтобы разрешить это (в частности, связанные списки, которые предлагали постоянное время splice ()) , Таким образом, их разделение имело смысл. Я не думаю, что это несоответствие разрешено больше, как бы то ни было.
DrPizza

Я согласен. C ++ рассматривает std::map<T,U>как контейнер пар. В Python a dictдействует как его ключи при повторении, но позволяет вам d.items()получить поведение C ++. Python также предоставляет d.values(). std::map<T,U>безусловно , может обеспечить keys()и values()метод , который возвращает объект , который имеет begin()и end()которые обеспечивают итераторы над ключами и значениями.
Бен
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.