如何使用多个键对std :: set进行排序

时间:2012-09-25 08:35:36

标签: c++ sorting set

我需要根据多个键将一个集复制到另一个集。 这些键用于 - 集体保持集合中元素的唯一性和顺序。

我的课程:

class LaneConnector {
public:

    const Lane* getLaneFrom() const {
        return From;
    }
    const Lane* getLaneTo() const {
        return To;
    }

private:

    Lane* From;
    Lane* To;
}

我的算子:

struct MyLaneConectorSorter {
  bool operator() (const LaneConnector* rhs, const LaneConnector* lhs) const
  {

    const Lane* a = lhs->getLaneFrom();
    const Lane* b = rhs->getLaneFrom();

    bool key1 = a->getLaneID() < b->getLaneID();
    bool key2 = a->getLaneParent->ID() < b->getLaneParent->ID();
    bool key2 = a->getLaneParent->getParent->ID() < b->getLaneParent->getParent->ID(); 
    //remind you that I NEED the elements to be in ascending order of 
    //getLaneParent->getParent->ID() ,a->getLaneParent->ID() and then a->getLaneID()
    //duplicate elements are the ones which have all three keys same and need to be discarded 
    return (key1 && key2 && key3); //which dont seem to be working
  }
};

和我的来源和原点设置:

const std::set<LaneConnector*> src = ..... ; //the getter give me a const version
std::set<sim_mob::LaneConnector *, MyLaneConectorSorter> dest;

以及我如何填写:

for(std::set<sim_mob::LaneConnector*>::iterator it = tempLC.begin(); it != tempLC.end(); it++)
{
    dest.insert(*it);//I know I can insert it right at the time of declaration, but keep it like this for now...please 
}

您的帮助将受到高度赞赏。

5 个答案:

答案 0 :(得分:3)

你需要优先考虑你的关键字段比较......只有当最重要的字段相等时,你才能比较第二个最重要的字段 - 如果它相等则你比较第三个最重要的字段等等。一旦存在不平等,你适当地返回true或false。因此,它不是&&操作,应该是? :if - else链,如:

return lhs.key1 < rhs.key1 ? true :
       rhs.key1 < lhs.key1 ? false :
       lhs.key2 < rhs.key2 ? true :
       rhs.key2 < lhs.key2 ? false :
       ...
       false;

要使设置正常运行,您必须确保密钥永远不会相等 - 以便永远不会使用最后false

答案 1 :(得分:3)

由于正确地为多个测试获取operator<非常困难,所以我提倡my way of doing this with tuple(在这种情况下使用make_tuple代替tie,因为我们正在处理已退回的临时工具来自功能):

#include <tuple>

struct MyLaneConectorSorter {
  bool operator() (const LaneConnector* lhs, const LaneConnector* rhs) const
  {
    const Lane* a = lhs->getLaneFrom();
    const Lane* b = rhs->getLaneFrom();
    auto const* pa = a->getLaneParent();
    auto const* pb = b->getLaneParent();

    return std::make_tuple(a->getLaneID(), pa->ID(), pa->getParent()->ID()) < 
           std::make_tuple(b->getLaneID(), pb->ID(), pb->getParent()->ID())
}

如果你的编译器还没有提供它们,你也可以从Boost获得tuplemake_tuple

答案 2 :(得分:2)

如果您有三个成员foo,bar和baz进行比较,这是比较它们的常用方法:

return lhs.foo < rhs.foo
    || lhs.foo == rhs.foo && (lhs.bar < rhs.bar
                           || lhs.bar == rhs.bar && lhs.baz < rhs.baz);

你看到了这种模式吗? ;)

答案 3 :(得分:1)

我在理解你的排序规则时遇到了问题,但是如果关系是一个简单的子排序,那么代码应该是这样的:

if (a->getLaneID() < b->getLaneID())
  return true;
else if (a->getLaneID() == b->getLaneID())
{
  if (a->getLaneParent->ID() < b->getLaneParent->ID())
    return true;
  // etc...
}
return false;

答案 4 :(得分:1)

您的class MyLaneConnectionSorter有缺陷。

std::set期望一个可以对元素进行排序的比较类。因此,您的比较函数必须提供类似于less仿函数或operator<的行为,即a < ba > bb < a)或a == b!(a < b) && !(a > b)

如果我们采用你的比较函数,它将考虑Lanes(6,5,4)和(7,3,4)(格式(PPID,PID,ID))相等,因为两者都不小于另一个。所以你需要像这样比较:

if (a->getLaneParent->getParent->ID() < b->getLaneParent->getParent->ID()) return true;
else if (a->getLaneParent->getParent->ID() > b->getLaneParent->getParent->ID()) return false;
else {
    if (a->getLaneParent->ID() < b->getLaneParent->ID()) return true;
    else if (a->getLaneParent->ID() > b->getLaneParent->ID()) return false;
    else {
        return (a->getLaneID() < b->getLaneID());
    }
}