如何在 std::set 中为唯一元素插入提供比较功能？

Question

我有以下用户定义的类，

class simple
{
public:
  simple(string str1, string str2)
   : s1(str1)
   , s2(str2)
  {
  }
  const string& getS1() const {return s1;}
  const string& getS2()const {return s2;} 
  bool operator<(const simple& s) {
    return strcmp(this->getS1().c_str(), s.getS1().c_str()) < 0 && strcmp(this->getS2().c_str(), s.getS2().c_str()) < 0;
  }
private:
  const string s1;
  const string s2;
}

我将这个类存储在 std::set 中，如下所示：

std::set<simple*> mySimplePtrSet;
simple* s1 = new simple("stack", "overflow");
simple* s2 = new simple("go", "ogle");
simple* s3 = new simple("stack", "overflow");
simple* s4 = new simple("go", "good");
simple* s5 = new simple("my", "overflow");

在上面的示例中，集合应该只包含 s1、s2、s4 和 s5。

考虑到类的数据成员（即s1和s2），如何在集合中存储唯一的类对象？

我尝试了以下比较器功能，但它不起作用。

struct myCom {
   bool operator()(const simple* a,const simple* b){
      return *a < *b;
   }
};

注意：在我的情况下，排序并不重要，如果插入未排序的元素对我来说没问题。

Answer 1

您正在存储指针，因此比较器只是比较内存地址，而不是您的自定义 operator<。

停止使用new并将实际对象存储在地图中：

std::set<simple> mySimpleSet;
mySimpleSet.emplace("stack", "overflow");
mySimpleSet.emplace("go", "ogle");
mySimpleSet.emplace("stack", "overflow");
mySimpleSet.emplace("go", "good");
mySimpleSet.emplace("my", "overflow");

或者如果您真的需要存储指针，那么您必须提供一个自定义比较器，该比较器不比较内存地址，而是比较这些内存地址处的对象。

您的operator< 也有缺陷。您正在使用std::string，因此请使用std::string::compare 而不是strcmp，并且您没有使用参数s。这样的事情会更好：

bool operator<(const simple& s) {
  return getS1() < s.getS1() && getS2() < s.getS2();
}