按照某个数据成员对STL结构体容器进行排序

3

我知道使用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)。我尝试在宏中用BOOST_TYPEOF(CONTAINER)::value_type替换TYPE,但编译器不喜欢这样做。
是否有办法在C++98中实现这一点,而不需要指定类型?另外,如果有一些方法可以减少甚至消除C++98中宏的使用,则更好了。

@JerryCoffin 这样做不行,因为CONTAINER是容器的一个实例而不是类型本身。 - Dave Johansen
糟糕 - 我应该更仔细地阅读。对不起。 - Jerry Coffin
2个回答

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将推导为fooR将推导为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;} );
}

这就是诀窍。它仍然需要在容器中指定类型(即foo),但它可以在没有宏的情况下工作,我会接受它。此外,在命名为sort_by和使用>运算符方面做得很好,当我组合示例时,使用>运算符是一个复制和粘贴错误。 - Dave Johansen

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

我同意那个方法是可行的,但它需要很多“样板文件”(比较函数的定义和使用.begin()/.end())。我希望在C++98/03代码中有一个不需要函数定义且接口尽可能简单的解决方案。 - Dave Johansen

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接