由某个数据成员对结构的STL容器进行排序

时间:2015-02-06 15:36:31

标签: c++ stl containers

我知道使用C ++ 11会更简单,但有没有办法让它在C ++ 98中运行?我有以下示例代码:

#include <boost/lambda/lambda.hpp>
#include <boost/lambda/bind.hpp>
#include <vector>
#include <algorithm>
#include <iostream>

struct foo {
  foo(const int a_, const double b_): a(a_), b(b_) {}
  int a;
  double b;
};

std::ostream &operator<<(std::ostream &out, const foo &f)
{
  out << f.a << ',' << f.b;
  return out;
}

#define DO_SORT(CONTAINER, TYPE, MEMBER) \
  std::sort(CONTAINER.begin(), CONTAINER.end(), boost::lambda::bind(&TYPE::MEMBER, boost::lambda::_1) > boost::lambda::bind(&TYPE::MEMBER, boost::lambda::_2));

int main()
{
  std::vector<foo> v;
  v.push_back(foo(1, -2.0));
  v.push_back(foo(3, -3.0));
  v.push_back(foo(2, -1.0));

  std::cout << "Original" << std::endl;
  std::for_each(v.begin(), v.end(), std::cout << boost::lambda::_1 << "\n");
  std::cout << "Sort by a" << std::endl;
  DO_SORT(v, foo, a);
  std::for_each(v.begin(), v.end(), std::cout << boost::lambda::_1 << "\n");
  std::cout << "Sort by b" << std::endl;
  DO_SORT(v, foo, b);
  std::for_each(v.begin(), v.end(), std::cout << boost::lambda::_1 << "\n");
}

这可以完成工作,但它需要指定容器内部的类型。我正在寻找一种能够只写DO_SORT(v, a)的方法。我尝试用TYPE替换宏中的BOOST_TYPEOF(CONTAINER)::value_type,但编译器并不喜欢这样。

有没有办法在C ++ 98中执行此操作而不需要指定类型?另外,如果有某种方法可以减少甚至删除C ++ 98中宏的使用,那么这也很好。

2 个答案:

答案 0 :(得分:2)

您可以使用模板参数推导消除宏并使用指向成员变量的指针获取类型:

template <typename C, typename T, typename R>
void do_sort(C& container, R T::*mem) {
    std::sort(
        container.begin(), container.end(), 
        boost::lambda::bind(mem, boost::lambda::_1) > 
            boost::lambda::bind(mem, boost::lambda::_2));
}

然后使用语法

调用
do_sort(v, &foo::a);

因此将实例化:

void do_sort(std::vector<foo>& container, int foo::*mem);

T将推断为foo,而R将推断为foo::a被声明为的任何内容,以便为指针到成员参数生成正确的类型。附注:我将此功能命名为sort_by。通常的做法也是使用<进行排序而不是>,但如果您想要降序值,那么这就是最佳选择。


为了完整起见,我将在C ++ 11中实现这一点

template <typename C, typename T, typename R>
void sort_by(C& container, R T::*mem) {
    std::sort(std::begin(container), std::end(container),
            [mem](const T& lhs, const T& rhs) {return lhs.*mem > rhs.*mem;} );
}

答案 1 :(得分:0)

您是仅仅将此作为一般情况的示例,还是会像这样工作?

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

struct foo {
  foo(const int a, const double b) : a(a), b(b) {}
  int a;
  double b;
};

bool compareA(const foo& foo1, const foo& foo2)
{
    return foo1.a < foo2.a;
}

bool compareB(const foo& foo1, const foo& foo2)
{
    return foo1.b < foo2.b;
}

int main()
{
  std::vector<foo> v;
  v.push_back(foo(1, -2.0));
  v.push_back(foo(3, -3.0));
  v.push_back(foo(2, -1.0));

  std::cout << "Original" << std::endl;
  for ( int i = 0; i < v.size(); i++ ) std:: cout << v[i].a << '\n';
  std::cout << "Sort by a" << std::endl;
  std::sort( v.begin(), v.end(), compareA );
  for ( int i = 0; i < v.size(); i++ ) std:: cout << v[i].a << '\n';
  std::cout << "Sort by b" << std::endl;
  std::sort( v.begin(), v.end(), compareB );
  for ( int i = 0; i < v.size(); i++ ) std:: cout << v[i].b << '\n';
}