C ++比较共享指针的堆栈

时间:2019-02-19 05:37:06

标签: c++ pointers shared-ptr comparator

假设我有一个堆栈,其中包含int的共享指针,如下所示:

#include <stack>
#include <memory>

using namespace std;

int main()
{
    stack<shared_ptr<int>> s1;
    stack<shared_ptr<int>> s2;

    shared_ptr<int> v1 = make_shared<int>(1);
    shared_ptr<int> v2 = make_shared<int>(1);

    s1.push(v1);
    s2.push(v2);

    bool areEqual = s1 == s2; // This is false
}

如何使堆栈比较shared_ptr指向的实际值而不是指针本身?

2 个答案:

答案 0 :(得分:4)

std::stack具有受保护的成员c,该成员是基础容器类型的实例。您可以制作一个访问该变量的堆栈包装器,然后按如下方式比较基础容器的内容:

#include <iostream>
#include <stack>
#include <memory>
#include <algorithm>

using namespace std;

template<class stack_type>
struct stack_wrapper : stack_type
{
    auto begin() const
    {
        return stack_type::c.begin();
    }

    auto end() const
    {
        return stack_type::c.end();
    }
};

template<class stack_type>
const stack_wrapper<stack_type> &wrap(const stack_type &stack)
{
    return static_cast<const stack_wrapper<stack_type> &>(stack);
}

int main() {
    stack<shared_ptr<int>> s1;
    stack<shared_ptr<int>> s2;

    shared_ptr<int> v1 = make_shared<int>(1);
    shared_ptr<int> v2 = make_shared<int>(1);

    s1.push(v1);
    s2.push(v2);

    const auto &s1wrapper = wrap(s1);
    const auto &s2wrapper = wrap(s2);

    const auto is_equal = std::equal(s1wrapper.begin(),
        s1wrapper.end(),
        s2wrapper.begin(),
        s2wrapper.end(),
        [](auto &first, auto &second) {
            return first && second && *first == *second;
        });

    std::cout << is_equal << std::endl;
}

答案 1 :(得分:3)

我喜欢@RealFresh's answer。它确切地说明了受保护的可访问性实际上是如何“封装”的。但是,将基类子对象引用转换为派生类子对象引用,然后将其视为一个对象,会很快导致不确定的行为。

但是提取c成员的想法很合理。我们可以使用简单的实用程序来做到这一点,而不会导致UB风险:

template<class S>
constexpr decltype(auto) stack_c(S&& s) {
    using base = std::decay_t<S>;
    struct extractor : base {
        using base::c;
    };
    constexpr auto c_ptr = &extractor::c;
    return std::forward<S>(s).*c_ptr;
} 

由于表达式&extractor::c works的使用方式,我们实际上获得了一个指向base(专门化std::stack)的成员c的指针。 。 extractor的目的是通过使用声明使该名称可公开访问。

然后,我们转发对它的引用,保留值类别等等。这是 @RealFresh的使用std::equal的建议中的替代项:

bool areEqual = std::equal(
    stack_c(s1).begin(), stack_c(s1).end(),
    stack_c(s2).begin(), stack_c(s2).end(),
    [](auto const& p1, auto const& p2) {
        return first && second && (p1 == p2 || *p1 == *p2);
    }
); 

See it live