使用带有boost :: shared_ptr的基类在地图中查找

时间:2014-05-07 13:26:09

标签: c++ boost shared-ptr

我正在寻找一种使用基类在地图中查找元素的方法(下面的代码只是一个基本示例):

#include <map>
#include <boost/shared_ptr.hpp>

class Base {
public:
    Base(int v) : id(v) {};
    int id;  
};

class Derived : public Base {
public:
    Derived(int v) : Base(v) {};
};


int main()
{
    std::map<boost::shared_ptr<Derived>, double> m;
    m.insert(std::make_pair(boost::shared_ptr<Derived>(new Derived(1)), 10));
    m.insert(std::make_pair(boost::shared_ptr<Derived>(new Derived(2)), 20));

    auto b1 = boost::shared_ptr<Base>(new Base(1));
    m.find(b1);

    return 0;
}

基本上,我想比较id属性。编译器返回的错误如下:

main.cpp: In function 'int main()':
main.cpp:35:14: error: no matching function for call to 'std::map<boost::shared_ptr<Derived>, double>::find(boost::shared_ptr<Base>&)'
     m.find(b1);
              ^
main.cpp:35:14: note: candidates are:
In file included from /usr/include/c++/4.8/map:61:0,
                 from main.cpp:1:
/usr/include/c++/4.8/bits/stl_map.h:820:7: note: std::map<_Key, _Tp, _Compare, _Alloc>::iterator std::map<_Key, _Tp, _Compare, _Alloc>::find(const key_type&) [with _Key = boost::shared_ptr<Derived> _Tp = double; _Compare = std::less<boost::shared_ptr<Derived> > _Alloc = std::allocator<std::pair<const boost::shared_ptr<Derived>, double> > std::map<_Key, _Tp, _Compare, _Alloc>::iterator = std::_Rb_tree_iterator<std::pair<const boost::shared_ptr<Derived>, double> > std::map<_Key, _Tp, _Compare, _Alloc>::key_type = boost::shared_ptr<Derived>]
       find(const key_type& __x)
       ^
/usr/include/c++/4.8/bits/stl_map.h:820:7: note:   no known conversion for argument 1 from 'boost::shared_ptr<Base>' to 'const key_type& {aka const boost::shared_ptr<Derived>&}'
/usr/include/c++/4.8/bits/stl_map.h:835:7: note: std::map<_Key, _Tp, _Compare, _Alloc>::const_iterator std::map<_Key, _Tp, _Compare, _Alloc>::find(const key_type&) const [with _Key = boost::shared_ptr<Derived> _Tp = double; _Compare = std::less<boost::shared_ptr<Derived> > _Alloc = std::allocator<std::pair<const boost::shared_ptr<Derived>, double> > std::map<_Key, _Tp, _Compare, _Alloc>::const_iterator = std::_Rb_tree_const_iterator<std::pair<const boost::shared_ptr<Derived>, double> > std::map<_Key, _Tp, _Compare, _Alloc>::key_type = boost::shared_ptr<Derived>]
       find(const key_type& __x) const
       ^
/usr/include/c++/4.8/bits/stl_map.h:835:7: note:   no known conversion for argument 1 from 'boost::shared_ptr<Base>' to 'const key_type& {aka const boost::shared_ptr<Derived>&}'

5 个答案:

答案 0 :(得分:3)

如果您希望使用地图进行id查找,则需要传入适当的比较函数,以便地图按id而不是默认operator <对其键进行排序(我相信,它会将所有权块地址与boost::shared_ptr个参数进行比较)。

所以改变这样的地图:

struct Less_id
{
  bool operator() (const boost::shared_ptr<Derived> &lhs, const boost::shared_ptr<Derived> &rhs) const
  {
    return lhs->id < rhs->id;
  }
};

typedef std::map<boost::shared_ptr<Derived>, double, Less_id> Map;

Map m;

这会相应地对地图进行排序,但仍然不允许按Base指针进行查找。为此,您可以在std::lower_bound上面编写自己的函数:

Map::const_iterator find_base(const Map &map, const boost::shared_ptr<Base> &base)
{
  auto it = std::lower_bound(
    map.begin(), map.end(), base,
    [](const Map::value_type &lhs, const boost::shared_ptr<Base> &rhs)
    { return lhs.first->id < rhs->id; }
  );
  if (it != map.end() && it->first->id == base->id)
    return it;
  else
    return map.end();
}

std::lower_bound()用于保持std::map::find()提供的对数复杂度。

Live example

答案 1 :(得分:1)

有两个问题需要克服。首先,您要在Base的集合中搜索Derived。第二个是你想要按值而不是按地址进行比较。 其他答案忽略了第二点。试试std::find_if

auto b1 = boost::shared_ptr<Base>(new Base(1));
auto itFound = std::find_if(m.begin(), m.end()
   [=](const std::pair<boost::shared_ptr<Derived>, double>& pair)
   {
      // omitting null checks for this example
      return pair.first->id == b1->id;
   });

如果要求实际上只是找到具有给定id的密钥,那么可以使其更简单:

int queryKey = 1;
auto itFound = std::find_if(m.begin(), m.end()
   [=](const std::pair<boost::shared_ptr<Derived>, double>& pair)
   {
      return pair.first->id == queryKey;
   });

现在,正如评论中所述,这将为您提供线性而非map常用的对数查找时间。如果地图很小,则无关紧要,但如果这是一个问题,您可以使用std::lower_bound代替find_if。请注意,这还需要添加自定义比较器,以确保映射的排序顺序基于id。例如:

struct Compare
{
   bool operator()(const boost::shared_ptr<Derived>& l,
      const boost::shared_ptr<Derived>& r) const
   {
      // omitting null checks for this example
      return l->id < r->id;
   }
};

std::map<boost::shared_ptr<Derived>, double, Compare> m;

答案 2 :(得分:1)

使用

std::map<boost::shared_ptr<Derived>, double, std::less<boost::shared_ptr<Base>>>

编辑:我领先于时代 - 这只适用于C ++ 14。

答案 3 :(得分:0)

这是因为boost :: shared_ptr的运算符&lt;不是你想要的,你需要一个委托给Derived类的运算符&lt;。

使用std :: map&lt; _Key,_Tp,_Compare&gt;

例如:

 std::map<boost::shared_ptr<Derived>, double, compare_func()> m;

答案 4 :(得分:0)

当您正在寻找一个只能在地图中Base的对象Derived时,您可以这样做:

boost::shared_ptr<Derived> d1 = boost::dynamic_pointer_cast<Derived>(b1);
if(d1) {
   m.find(d1);
} else {
   // not in the map
}

当您使用std::shared_ptr<Derived>作为时,另一种可能性是实际使用指向基类的指针:

使用

std::map<boost::shared_ptr<Base>, double> m;

而不是

std::map<boost::shared_ptr<Derived>, double> m;

一切都按预期工作。


顺便说一句,你在virtual中遗漏了Base析构函数!