Como recuperar todas as chaves (ou valores) de um std :: map e colocá-las em um vetor?

246

Esta é uma das maneiras possíveis de sair:

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"));

Obviamente, também podemos recuperar todos os valores do mapa, definindo outro functor RetrieveValues .

Existe alguma outra maneira de conseguir isso facilmente? (Estou sempre me perguntando por que std :: map não inclui uma função de membro para fazer isso.)

Owen
fonte
10
sua solução é a melhor ...
linello
4
A única coisa que eu gostaria de acrescentar t este é keys.reserve(m.size());.
Galik

Respostas:

176

Embora sua solução funcione, pode ser difícil de ler, dependendo do nível de habilidade de seus colegas programadores. Além disso, ele afasta a funcionalidade do site de chamada. O que pode dificultar um pouco a manutenção.

Não tenho certeza se o seu objetivo é colocar as chaves em um vetor ou imprimi-las para que eu faça as duas coisas. Você pode tentar algo como isto:

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";
}

Ou ainda mais simples, se você estiver usando o 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";
}

Pessoalmente, gosto da versão BOOST_FOREACH porque há menos digitação e é muito explícito sobre o que está fazendo.

Jere.Jones
fonte
1
Vá em números que eu acabaria aqui depois da minha pesquisa no Google. O seu é a resposta que eu prefiro :)
MPEN
4
@ Jere - Você realmente trabalhou BOOST_FOREACH? O código que você propõe aqui está totalmente errado
Manuel
2
@ Jamie - essa é outra maneira, mas os documentos de aumento mostram a especificação da variável e seu tipo antes do BOOST_FOREACH se o tipo contiver uma vírgula. Eles também mostram digitação. Então, estou confuso, o que há de errado com o meu código?
precisa saber é o seguinte
17
Curioso, não faria sentido presizer o vetor para impedir a alocação de redimensionamento?
19373 Alan
2
Não esqueça de v.reserve(m.size())evitar o redimensionamento do vetor durante a transferência.
Brian White
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);
Juan
fonte
4
Agradável. Esqueça it = ...begin(); it != ...end. Mais agradável seria, evidentemente, std :: map ter um chaves método () retornando que vector ...
masterxilo
2
@ BenHymers: Parece-me que esta resposta foi dada em answered Mar 13 '12 at 22:33, que é vários meses após o C ++ 11 se tornar C ++.
Sebastian Mach
37
@BenHymers, mas é útil para qualquer pessoa que esteja lendo a pergunta agora, que é a essência da SO - não apenas ajudando o solicitante, mas todos os outros.
Luchian Grigore
9
for (auto e imap) é mais preciso porque não há operação de cópia.
HelloWorld
2
@StudentT, melhor ainda for(auto const & imap : mapints),.
cp.engr 23/10/2015
61

Existe um adaptador de faixa de reforço para este fim:

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

Existe um adaptador de intervalo map_values ​​semelhante para extrair os valores.

Alastair
fonte
1
Infelizmente, parece que boost::adaptorsnão está disponível até o Boost 1.43. A atual versão estável do Debian (Squeeze) oferece apenas o Boost 1.42
Mickaël Le Baillif
2
É uma pena. O Boost 1.42 foi lançado em fevereiro de 2010, mais de 2,5 anos antes do Squeeze.
Alastair
Nesse ponto, o Squeeze Updates e o repositório de backports não deveriam oferecer o Boost 1.44?
Luis Machuca 28/03
em qual cabeçalho de impulso está definido?
James Wierzba
1
Veja o documento vinculado, está definido emboost/range/adaptor/map.hpp
Alastair
46

O C ++ 0x nos deu uma solução excelente e excelente:

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;});
DanDan
fonte
22
Na minha opinião, não há nada de excelente nisso. chaves std :: vector <int>; keys.reserve (m_Inputs.size ()); for (valor automático da chave: m_Inputs) {keys.push_back (valor da chave.first); } É muito melhor que a transformação enigmática. Mesmo em termos de desempenho. Este é melhor.
precisa
5
Você também pode reservar o tamanho das chaves aqui se desejar um desempenho comparável. use a transformação se desejar evitar um loop for.
DanDan
4
só quero adicionar - pode usar [] (const auto & pair)
ivan.ukr 2/16
@ ivan.ukr que compilador você está usando? Esta sintaxe não é permitida aqui: 'const auto': um parâmetro não pode ter um tipo que contenha 'auto'
Gobe
4
@ ivan.ukr parâmetro automático no lambda é c ++ 14
roalz 13/17
16

A resposta de @ DanDan, usando 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;
}); 

e usando C ++ 14 (como observado por ivan.ukr), podemos substituir decltype(map_in)::value_typepor auto.

James Hirschorn
fonte
5
Você pode adicionar keys.reserve(map_in.size());eficiência.
Galik 04/04
Acho que o método de transformação realmente leva mais código do que o loop for.
precisa saber é o seguinte
const pode ser colocado atrás do tipo! Eu quase esqueço isso.
Zhang
12

O SGI STL tem um ramal chamado select1st. Pena que não está no STL padrão!

Chris Jester-Young
fonte
10

Sua solução é boa, mas você pode usar um iterador para fazer isso:

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
}
Brian R. Bondy
fonte
10

Baseado na solução @ rusty-parks, mas em c ++ 17:

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

for (const auto e [tecla, ignorada]: itens)
{
    itemKeys.push_back (key);
}
Madiyar
fonte
Eu não acho que std::ignorepossa ser usado em ligações estruturadas dessa maneira. Estou recebendo um erro de compilação. Deve ser suficiente usar apenas uma variável regular, por exemplo, ignoredque simplesmente não seja usada.
jb
1
@jb Obrigado. De fato, std::ignoredestina-se ao uso com, std::tiemas não com ligações estruturais. Eu atualizei meu código.
Madiyar 12/04
9

Eu acho que o BOOST_FOREACH apresentado acima é agradável e limpo, no entanto, há outra opção usando o BOOST também.

#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") );

Pessoalmente, não acho que essa abordagem seja tão limpa quanto a abordagem BOOST_FOREACH nesse caso, mas o boost :: lambda pode ser realmente limpo em outros casos.

paxos1977
fonte
7

Além disso, se você tiver o Boost, use transform_iterator para evitar fazer uma cópia temporária das chaves.

Marcelo Cantos
fonte
7

Um pouco de uma captura de c ++ 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;
}
Rusty Parks
fonte
5

Aqui está um bom modelo de função usando magia C ++ 11, trabalhando para 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;
}

Confira aqui: http://ideone.com/lYBzpL

Clemens Sielaff
fonte
4

A melhor solução STL sem sgi e sem boost é estender o map :: iterator da seguinte forma:

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());
}

e depois use-os assim:

        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];
Marius
fonte
1
Vou deixar para o leitor também criar o const_iterator e reverter os iteradores, se / quando necessário.
Marius
-1

Com exemplo de mapa atômico

#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;
}
Deniz Babat
fonte
-2

Um pouco semelhante a um dos exemplos aqui, simplificado da std::mapperspectiva do uso.

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;
}

Use assim:

auto keys = getKeys(yourMap);
TarmoPikaro
fonte
2
Ei, eu sei que essa resposta é antiga, mas também está errada. Inicializar com tamanho map.size()significa o dobro do retorno do tamanho do vetor. Por favor, corrija para salvar outra pessoa da dor de cabeça :(
thc
-3

(Estou sempre me perguntando por que std :: map não inclui uma função de membro para fazer isso.)

Porque ele não pode fazer isso melhor do que você pode fazê-lo. Se a implementação de um método não for superior à implementação de uma função livre, em geral você não deve escrever um método; você deve escrever uma função livre.

Também não está claro imediatamente por que ele é útil.

DrPizza
fonte
8
Há outros motivos além da eficiência para uma biblioteca fornecer um método, como a funcionalidade "baterias incluídas" e uma API encapsulada e coerente. Embora admitidamente nenhum desses termos descreva o STL particularmente bem :) Re. não está claro por que é útil - realmente? Eu acho que é bastante óbvio por que listar as chaves disponíveis é uma coisa útil para poder fazer com um mapa / dicionário: depende do que você está usando.
andybuckley
4
Por esse raciocínio, não deveríamos ter, empty()porque pode ser implementado como size() == 0.
Gd1
1
O que @ gd1 disse. Embora não deva haver muita redundância funcional em uma classe, insistir em absolutamente zero não é uma boa idéia IMO - pelo menos até que o C ++ permita "abençoar" as funções livres nos métodos.
einpoklum
1
Nas versões mais antigas do C ++, havia contêineres para os quais vazio () e tamanho () podiam razoavelmente ter garantias de desempenho diferentes, e acho que a especificação estava suficientemente frouxa para permitir isso (especificamente, listas vinculadas que ofereciam emenda de tempo constante ()) . Como tal, dissociá-los fazia sentido. Eu não acho que essa discrepância seja mais permitida, no entanto.
DrPizza
Concordo. O C ++ trata std::map<T,U>como um contêiner de pares. No Python, a dictage como suas chaves quando iterado, mas permite que você d.items()obtenha o comportamento do C ++. Python também fornece d.values(). std::map<T,U>certamente poderia fornecer um método keys()e values()que retornam um objeto que possui begin()e end()que fornecem iteradores sobre as chaves e valores.
Ben