使用operator ==时std :: set中unique_ptr的深度比较

时间:2014-08-28 13:42:29

标签: c++ c++11

我正在尝试使用std::set将一组unique_ptr保存到我定义的自定义对象中。我在定义集合时提供自定义比较功能(以启用深度比较)。当将元素插入集合中时,此比较函数似乎正常工作,即具有相同内容的项目未插入两次。

但是,如果我使用operator==比较两个集合,它似乎被忽略,即具有等效元素的集合返回为不相等,而我期望(希望)它相等(作为自定义比较函数)我提供了深入的比较)。

比较功能是否仅在插入期间使用?如果是这样,是否有替代方法让operator==进行深入比较?

任何指针都表示赞赏。谢谢:))

示例代码

//
//  main.cpp
//  Test

#include <iostream>
#include <set>

class Person {
private:
    std::string mName;

public:
    Person(const std::string& name);
    virtual ~Person() {}

    void setName(std::string& name);
    std::string getName();
};

typedef std::unique_ptr<Person> PersonUniquePtr;

Person::Person(const std::string& name)
    : mName{ name }
{
}

void Person::setName(std::string& name)
{
    mName = name;
}   

std::string Person::getName()
{
    return mName;
}

bool isLess(Person* p1, Person* p2)
{
    if (p1->getName().compare(p2->getName()) == -1)
        return true;

    return false;
}

struct PersonUniquePtr_less {
    bool operator()(PersonUniquePtr const& p1, PersonUniquePtr const& p2) const
    {
        return isLess(p1.get(), p2.get());
    }
};

int main(int argc, const char* argv[])
{
    std::set<PersonUniquePtr, PersonUniquePtr_less> personSet1;
    std::set<PersonUniquePtr, PersonUniquePtr_less> personSet2;

    PersonUniquePtr person1 = std::make_unique<Person>("Adam");
    PersonUniquePtr person2 = std::make_unique<Person>("Adam");
    personSet1.insert(std::move(person1));
    personSet1.insert(std::move(person2));
    std::cout << "personSet1.size(): " << personSet1.size() << std::endl; //Expected 1

    PersonUniquePtr person3 = std::make_unique<Person>("Bruce");
    personSet1.insert(std::move(person3));
    std::cout << "personSet1.size(): " << personSet1.size() << std::endl; //Expected 2

    PersonUniquePtr person4 = std::make_unique<Person>("Adam");
    PersonUniquePtr person5 = std::make_unique<Person>("Bruce");
    personSet2.insert(std::move(person4));
    personSet2.insert(std::move(person5));
    std::cout << "personSet2.size(): " << personSet2.size() << std::endl; //Expected 2

    std::cout << "PersonSet1:" << std::endl;
    for (auto& person : personSet1) {
        std::cout << person->getName() << std::endl;
    } //Prints out Adam Bruce

    std::cout << "PersonSet2:" << std::endl;
    for (auto& person : personSet2) {
        std::cout << person->getName() << std::endl;
    } //Prints out Adam Bruce

    bool setsAreEqual = (personSet1 == personSet2);
    if (setsAreEqual) {
        std::cout << "Sets are equal" << std::endl;
    } else {
        std::cout << "Sets are not equal" << std::endl;
    }

    return 0;
}

1 个答案:

答案 0 :(得分:7)

C ++ 11容器要求说a == b等同于

distance(a.begin(), a.end()) == distance(b.begin(), b.end())
&& equal(a.begin(), a.end(), b.begin())

std::equal未使用您的自定义比较,它使用operator==

您可以使用自定义谓词调用std::equal来自行执行比较:

a.size() == b.size()
&& std::equal(a.begin(), a.end(), b.begin(),
              [](PersonUniquePtr const& p1, PersonUniquePtr const& p2) {
                PersonUniquePtr_less cmp;
                return !cmp(p1, p2) && !cmp(p2, p1);
              });

在C ++ 14中,它更简单,因为std::equal的新重载占用了四个迭代器,尽管正如TemplateRex在下面的注释中指出的那样,这比测试{{{1}更低效。 1}}手动:

a.size() == b.size()

在C ++ 14中,您可以使用多态lambda来保存一些输入:

std::equal(a.begin(), a.end(), b.begin(), b.end(),
           [](PersonUniquePtr const& p1, PersonUniquePtr const& p2) {
             PersonUniquePtr_less cmp;
             return !cmp(p1, p2) && !cmp(p2, p1);
           });
相关问题