我希望std::set
如果shared_ptr's
比较的是指针,而不是指针。
我有这样一个例子:
std::shared_ptr<std::string> s(new std::string("abc"));
std::shared_ptr<std::string> p(new std::string("abc"));
std::set<std::shared_ptr<std::string>> S;
S.insert(s);
S.insert(p);
std::cout << S.size();
正如你所看到的,我在set
中放入了相同的元素,但这输出2。
如何使set的插入使用底层字符串的比较标准?如果它不是字符串而是更复杂的对象呢?
std::set
的第二个模板参数是要使用的比较器的类型(默认为std::less<Key>
):
#include <iostream>
#include <memory>
#include <set>
#include <string>
struct deref_less {
bool operator()(const auto& a, const auto& b) const { return (*a) < (*b); }
using is_transparent = void;
};
int main() {
std::shared_ptr<std::string> s(new std::string("abc"));
std::shared_ptr<std::string> p(new std::string("abc"));
std::set<std::shared_ptr<std::string>, deref_less> S;
S.insert(s);
S.insert(p);
std::cout << S.size();
}
输出:
1
auto
参数是为了方便使用c++ 20,之前比较器有点冗长。using is_transparent = void;
使set::find
过载,例如接受std::unique_ptr<std::string>
(参见godbolt的例子)。