2012-06-15 5 views
5

Voglio ottenere gli elementi ordinati per il numero della loro occorrenza. Questo è quello che ho messo a punto (mHeights è uno std :: multiset):Come ordinare un multiset a un contenitore per il numero di occorrenze dell'elemento

namespace{ 
    template<class U,class T> 
    class HistPair{ 
    public: 
     HistPair(U count,T const& el):mEl(el),mNumber(count){  
     } 
     T const& getElement()const{return mEl;} 

     U getCount()const{return mNumber;} 
    private: 
     T mEl; 
     U mNumber; 
    }; 

    template<class U,class T> 
    bool operator <(HistPair<U,T> const& left,HistPair<U,T> const& right){ 
    return left.getCount()< right.getCount(); 
    } 
} 

std::vector<HistPair<int,double> > calcFrequentHeights(){ 
    typedef HistPair<int,double> HeightEl; 
    typedef std::vector<HistPair<int,double> > Histogram; 
    std::set<double> unique(mHeights.begin(),mHeights.end()); 
    Histogram res; 
    boostForeach(double el, unique) { 
    res.push_back(HeightEl(el,mHeights.count(el)));  
    } 
    std::sort(res.begin(),res.end()); 
    std::reverse(res.begin(),res.end()); 
    return res; 
} 

Quindi, prima di iniziare tutti gli elementi unici dal multiset, poi li conto e ordinarli in un nuovo contenitore (I ho bisogno dei conteggi quindi uso una mappa). Questo sembra abbastanza complicato per un compito così facile. Oltre alla HistPair, che è usata anche altrove, non esiste alcun algoritmo stl che possa semplificare questa attività, ad es. usando equal_range o sth. nello stesso modo.

Edit: mi serve il numero di occorrenze così, mi dispiace dimenticato che

+1

sembra abbastanza succinta a me . Non riesco a immaginare che tu stia pianificando più di una linea o due dalla tua routine di creazione dell'istogramma. Un'alternativa potrebbe essere usare un 'std :: map ' invece di 'std :: multiset ' e calcolare le dimensioni del contenitore mentre si inseriscono gli elementi, ma non cambierà molto. – Rook

+0

@nims: Questo è quello che sto facendo o ti ho frainteso. Ho bisogno del conteggio e dell'elemento – Martin

+0

@Martin mio cattivo, l'ho trascurato. – nims

risposta

6

Questo frammento fa ciò che si vuole, combinando un std::set, un lambda e std::multiset::count:

#include <iostream> 
#include <set> 
#include <vector> 
#include <algorithm> 

int main() { 
    std::multiset<int> st; 
    st.insert(12); 
    st.insert(12); 
    st.insert(12); 
    st.insert(145); 
    st.insert(145); 
    st.insert(1); 
    st.insert(2); 

    std::set<int> my_set(st.begin(), st.end()); 
    std::vector<int> my_vec(my_set.begin(), my_set.end()); 
    std::sort(my_vec.begin(), my_vec.end(), 
     [&](const int &i1, const int &i2) { 
      return st.count(i1) < st.count(i2); 
     } 
    ); 

    for(auto i : my_vec) { 
     std::cout << i << " "; 
    } 
    std::cout << std::endl; 
} 

Potrebbero voglio invertire il vettore. Questo Uscite:

1 2 145 12 

Edit: Tenendo conto è necessario anche il conteggio degli elementi, questo lo farà:

#include <iostream> 
#include <set> 
#include <vector> 
#include <algorithm> 

int main() { 
    typedef std::vector<std::pair<int, int>> MyVector; 
    std::multiset<int> st; 
    st.insert(12); 
    st.insert(12); 
    st.insert(12); 
    st.insert(145); 
    st.insert(145); 
    st.insert(1); 
    st.insert(2); 

    std::set<int> my_set(st.begin(), st.end()); 
    MyVector my_vec; 
    my_vec.reserve(my_set.size()); 

    for(auto i : my_set) 
     my_vec.emplace_back(i, st.count(i)); 

    std::sort(my_vec.begin(), my_vec.end(), 
     [&](const MyVector::value_type &i1, const MyVector::value_type &i2) { 
      return i1.second < i2.second; 
     } 
    ); 

    for(const auto &i : my_vec) 
     std::cout << i.first << " -> " << i.second << std::endl; 
} 

quali uscite:

1 -> 1 
2 -> 1 
145 -> 2 
12 -> 3 
+0

Grazie. Lambdas è solo a portata di mano ;-) Tuttavia ho dimenticato di dire che ho bisogno anche dei conteggi, ma questo non dovrebbe essere difficile da integrare nella tua soluzione – Martin

+0

Ecco, hai risolto la mia risposta. – mfontanini

+0

Grazie ancora !!! – Martin

Problemi correlati