Question

我遇到的问题是，我有一个实现了shared_ptr的自定义类的向量operator<。

在堆栈上使用该类时，可以使用std :: sort而不指定Compare作为sort参数。但是，当然，当我使用shared_ptr时（很显然，我认为）会尝试对指针而不是对象本身进行排序。

我想知道是否可以在具有智能指针的容器上调用std::sort，并仍然可以调用实际对象的Compare运算符，而不是不提供的指针“ sort”的第三个参数。

为确保完整性，MWE：

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

class Date
{
public:
    Date(int y, int m) : year(y), month(m) {}
    bool operator<(const Date& rhs)
    {
        if (year > rhs.year) return false;
        else if (year == rhs.year)
        {
            return month < rhs.month;
        }
        return true;
    }
    int year;
    int month;
};

int main()
{
    Date d1(1999,12);
    Date d3(2000,10);
    Date d2(2000,1);
    Date d4(1997,9);

    std::vector<std::shared_ptr<Date>> dates = {std::make_shared<Date>(d1), std::make_shared<Date>(d2), std::make_shared<Date>(d3), std::make_shared<Date>(d4)};

    std::sort(dates.begin(), dates.end()); // doesn't work. orders by pointers

    // std::sort(dates.begin(), dates.end(), [](std::shared_ptr<Date> d1, std::shared_ptr<Date> d2) { return *d1 < *d2; }); // this works

    for (const auto& d : dates)
    {
        std::cout << d->year << " " << d->month << '\n';
    }
    return 0;
}

Answer 1

您已经发现，std::shared_ptr的比较运算符使用它所引用的指针。这样，分配的2个比较相等的实例仍然比较不相等。

使用函数显式排序是一件好事。

std::sort(dates.begin(), dates.end(), [](std::shared_ptr<Date> d1, std::shared_ptr<Date> d2) { return *d1 < *d2; });

但是，如果必须在多个位置进行操作，则可以将ptr包装在类/结构中：

template<typename T>
struct SharedPtr final
 {
       std::shared_ptr<T> ptr;
       bool operator==(const T &rhs) const
        { return *ptr == rhs; }
       bool operator==(const SharedPtr<T> &rhs) const
        { return *ptr == *rhs.ptr; }
      // ...
 };

可以使用nullptr检查，其他重载和运算符随意扩展<< / p>

Answer 2

在您的情况下，需要提供第3个参数。但是，如果您打算大量使用它，则可以通过创建一个特殊的结构来简化它，该结构会重载operator()

class Date
{
public:
    struct CompareSharedPtr
    {
        bool operator()(const std::shared_ptr<Date>& d1, const std::shared_ptr<Date>& d2)
        {
            return *d1 < *d2;
        }
    };
    Date(int y, int m) : year(y), month(m) {}
    bool operator<(const Date& rhs)
    {
        if (year > rhs.year) return false;
        else if (year == rhs.year)
        {
            return month < rhs.month;
        }
        return true;
    }
    int year;
    int month;
};

和用法

std::sort(dates.begin(), dates.end(), Date::CompareSharedPtr{});

使用std :: sort

2 个答案: