Sorting a vector of objects by a property of the object

后端 未结 4 682
春和景丽
春和景丽 2020-12-01 15:03

I\'m working on a project for school and need to sort some data. I\'ve been given a vector of objects and I have to sort the objects (either in place or using an index) base

相关标签:
4条回答
  • 2020-12-01 15:37

    There are several different objects and several different properties that could it be sorted by.

    While the solution Erik posted is correct, this statement leads me to think that it's impractical at best if you are in fact planning to sort by multiple public data members of multiple classes in multiple ways in the same program, as each sorting method will require its own functor type.

    I recommend the following abstraction:

    #include <functional>
    
    template<typename T, typename M, template<typename> class C = std::less>
    struct member_comparer : std::binary_function<T, T, bool>
    {
        explicit member_comparer(M T::*p) : p_(p) { }
    
        bool operator ()(T const& lhs, T const& rhs) const
        {
            return C<M>()(lhs.*p_, rhs.*p_);
        }
    
    private:
        M T::*p_;
    };
    
    template<typename T, typename M>
    member_comparer<T, M> make_member_comparer(M T::*p)
    {
        return member_comparer<T, M>(p);
    }
    
    template<template<typename> class C, typename T, typename M>
    member_comparer<T, M, C> make_member_comparer2(M T::*p)
    {
        return member_comparer<T, M, C>(p);
    }
    

    Usage would look like:

    #include <algorithm>
    #include <vector>
    #include <string>
    
    struct MyClass
    {
        int i;
        std::string s;
    
        MyClass(int i_, std::string const& s_) : i(i_), s(s_) { }
    };
    
    int main()
    {
        std::vector<MyClass> vec;
        vec.push_back(MyClass(2, "two"));
        vec.push_back(MyClass(8, "eight"));
    
        // sort by i, ascending
        std::sort(vec.begin(), vec.end(), make_member_comparer(&MyClass::i));
        // sort by s, ascending
        std::sort(vec.begin(), vec.end(), make_member_comparer(&MyClass::s));
        // sort by s, descending
        std::sort(vec.begin(), vec.end(), make_member_comparer2<std::greater>(&MyClass::s));
    }
    

    This will work for any type with public data members, and will save a lot of typing if you need to sort your classes more than a couple of different ways.

    Here is a variation that works with public member functions instead of public data members:

    #include <functional>
    
    template<typename T, typename M, template<typename> class C = std::less>
    struct method_comparer : std::binary_function<T, T, bool>
    {
        explicit method_comparer(M (T::*p)() const) : p_(p) { }
    
        bool operator ()(T const& lhs, T const& rhs) const
        {
            return C<M>()((lhs.*p_)(), (rhs.*p_)());
        }
    
    private:
        M (T::*p_)() const;
    };
    
    template<typename T, typename M>
    method_comparer<T, M> make_method_comparer(M (T::*p)() const)
    {
        return method_comparer<T, M>(p);
    }
    
    template<template<typename> class C, typename T, typename M>
    method_comparer<T, M, C> make_method_comparer2(M (T::*p)() const)
    {
        return method_comparer<T, M, C>(p);
    }
    

    With usage like:

    #include <algorithm>
    #include <vector>
    #include <string>
    
    class MyClass
    {
        int i_;
        std::string s_;
    
    public:
        MyClass(int i, std::string const& s) : i_(i), s_(s) { }
    
        int i() const { return i_; }
        std::string const& s() const { return s_; }
    };
    
    int main()
    {
        std::vector<MyClass> vec;
        vec.push_back(MyClass(2, "two"));
        vec.push_back(MyClass(8, "eight"));
    
        // sort by i(), ascending
        std::sort(vec.begin(), vec.end(), make_method_comparer(&MyClass::i));
        // sort by s(), ascending
        std::sort(vec.begin(), vec.end(), make_method_comparer(&MyClass::s));
        // sort by s(), descending
        std::sort(vec.begin(), vec.end(), make_method_comparer2<std::greater>(&MyClass::s));
    }
    
    0 讨论(0)
  • 2020-12-01 15:39

    Here is my version of the answer, just use a lambda function! It works, it uses way less code, and in my opinion it's elegant!

    #include <algorithm>
    #include <vector>
    #include <string>
    
    struct MyClass
    {
        int i;
        std::string s;
    
        MyClass(int i_, std::string const& s_) : i(i_), s(s_) { }
    };
    
    int main()
    {
        std::vector<MyClass> vec;
        vec.push_back(MyClass(2, "two"));
        vec.push_back(MyClass(8, "eight"));
    
        // sort by i, ascending
        std::sort(vec.begin(), vec.end(), [](MyClass a, MyClass b){ return a.i < b.i; });
        // sort by s, ascending
        std::sort(vec.begin(), vec.end(), [](MyClass a, MyClass b){ return a.s < b.s; });
        // sort by s, descending
        std::sort(vec.begin(), vec.end(), [](MyClass a, MyClass b){ return a.s > b.s; });
    }
    
    0 讨论(0)
  • 2020-12-01 15:49

    Use std::sort and a functor. e.g:

    struct SortByX
    {
       bool operator() const (MyClass const & L, MyClass const & R) { return L.x < R.x; }
    };
    
    std::sort(vec.begin(), vec.end(), SortByX());
    

    The functor's operator() should return true if L is less than R for the sort order you desire.

    0 讨论(0)
  • 2020-12-01 15:51

    EDIT: replacing with lambda_compare, because I'm a masochist:

    You can also just create a helper that lets you specify which value to use via a lambda expression:

    template <class F> class lambda_compare
    {
    public:
        lambda_compare(F f_): f(f_) { }
        template <class T> bool operator()(const T &lhs, const T &rhs) const
        { return f(lhs)<f(rhs); }
    
    private:
        F f;
    };
    
    template <class F> lambda_compare<F> make_lambda_compare(F f)
    { return f; }
    
    ...
    std::sort(vec.begin(), vec.end(), make_lambda_compare([](const foo &value) { return value.member; }));
    
    0 讨论(0)
提交回复
热议问题