首页 > 解决方案 > 在指针场景中使用 std::shared_ptr

问题描述

我创建了一个Node用于二叉搜索树实现的结构。它使用共享指针来跟踪它的孩子:

template <class T> struct Node;

template <class T>
using Node_ptr = std::shared_ptr<Node<T>>;

template <class T>
struct Node {
  Node_ptr<T> left;
  Node_ptr<T> right;
  const T label;
  ...
};

现在,我想要一个函数,它给出一些子树,一个值将返回该特定值的节点或该节点将来应该位于的位置 - find_node

这是它目前的样子:

template <class T>
auto* find_node(Node_ptr<T>* node, const T& value) {
  for (; *node && (*node)->label != value
       ; node = value < (*node)->label ? &(*node)->left : &(*node)->right);

  return node;
}

很糟糕。但它有效:

template <class T>
class Binary_search_tree {
public:
  // correctly inserts consecutive values
  void insert(const T& value) {
    if (auto* node = find_node(&root, value); !*node)
      *node = std::make_shared<Node<T>>(value);
  }
  ...
private:
  Node_ptr<T> root;
  ...
};

我可以重写find_node以使用std::shared_ptr<Node_ptr<T>>而不是,Node_ptr<T>*但它看起来会更糟。还是会?

我应该如何处理这种情况?

编辑:正如已经指出的那样,可以通过引用起始节点并返回对节点的引用来简化函数:

template <class T>
Node_ptr<T>& find_node(Node_ptr<T>& node_ref, const T& value) {
  auto* node = &node_ref;
  ...
  return *node;
}

标签: c++shared-ptrc++17smart-pointers

解决方案


  1. 当您必须允许传递空指针时,建议使用原始指针 - 在您的示例中不是这样;或者当您必须传递一个非整数值时(在您的情况下为 true)。在后一种情况下,仍应考虑传递引用而不是原始指针。这是一个通用建议 - 因此可能存在例外情况。

  2. 注意到这一点后,您仍然可以在这里相当安全地在函数中使用原始指针,方法是使 find_node(...) 成为私有函数,同时保持 insert(...) 公开。这是安全的,因为没有机会让指针从 insert(...) 内部悬空。

本质上,我们需要防止使用原始指针的两种可能性:#1。过早地删除指针指向的内存,#2。永远不要删除指针指向的内存。这在您的 insert(...) 函数中都是不可能的。所以你是安全的。

在相关说明中,您可能会考虑在创建节点时为节点设置 unique_pointer,然后如果它们要由多个子节点共享,则将它们转换为共享指针:std::move(...)。


推荐阅读