二叉搜索树--C++代码实现
2021/12/14 20:19:04
本文主要是介绍二叉搜索树--C++代码实现,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
typedef int DataType; struct TreeNode { DataType data; std::shared_ptr<TreeNode> left_node; std::shared_ptr<TreeNode> right_node; TreeNode() : data(0), left_node(nullptr), right_node(nullptr) {}; TreeNode(DataType value) : data(value), left_node(nullptr), right_node(nullptr) {}; }; class BinarySearchTree { public: BinarySearchTree() { root_tree_node_ = std::make_shared<TreeNode>(); } // 搜索 target 是否属于该树 bool Find(DataType target) { return Find(target, root_tree_node_); } bool Find(DataType target, std::weak_ptr<TreeNode> tree_node) { if(tree_node.expired()) { return false; } auto node_ptr = tree_node.lock(); if(node_ptr->data == target) { return true; } else if(node_ptr->data > target){ return Find(target, node_ptr->left_node); } else { return Find(target, node_ptr->right_node); } } //插入一个元素 void InsertData(DataType target) { auto insert_node = std::make_shared<TreeNode>(target); auto temp_node = root_tree_node_.get(); while(temp_node) { if(target < temp_node->data) { if(!temp_node->left_node) { temp_node->left_node = insert_node; num++; return; } else { temp_node = temp_node->left_node.get(); } } //对于相等的情况,按大于当前节点处理 else { if(!temp_node->right_node) { temp_node->right_node = insert_node; num++; return; } else { temp_node = temp_node->right_node.get(); } } } } //删除一个元素 bool DeleteData(DataType target) { if(!root_tree_node_) { return false; } auto tree_node = root_tree_node_; std::shared_ptr<TreeNode> node_base = nullptr; //找到待删除元素对应的节点 while(tree_node != nullptr && tree_node->data != target) { node_base = tree_node; if(tree_node->data < target) { tree_node = tree_node->right_node; } else { tree_node = tree_node->left_node; } } if(!tree_node) { return false; } //删除节点有两个子节点 if(tree_node -> left_node && tree_node->right_node) { std::shared_ptr<TreeNode> min_node = tree_node->right_node; std::shared_ptr<TreeNode> min_node_base = tree_node; //寻找右子树的最小节点,将它赋给当前待删除节点 while(min_node->left_node) { min_node_base = min_node; min_node = min_node->left_node; } tree_node->data = min_node->data; tree_node = min_node; node_base = min_node_base; } //删除节点是叶节点或仅有1个节点 std::shared_ptr<TreeNode> child_node; if(!tree_node -> left_node) { child_node = tree_node->left_node; } else if(!tree_node->right_node){ child_node = tree_node->right_node; } else { child_node = nullptr; } if(!node_base) { root_tree_node_ = child_node; } else if(tree_node == node_base->left_node) { node_base->left_node = child_node; } else node_base->right_node = child_node; return true; } private: int num = 1; std::shared_ptr<TreeNode> root_tree_node_; };
这篇关于二叉搜索树--C++代码实现的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-10-04el-table 开启定时器下,表格的选中状态会消失是什么原因-icode9专业技术文章分享
- 2024-10-03如何安装和初始化飞牛私有云 fnOS?-icode9专业技术文章分享
- 2024-10-03如何安装 App 并连接到飞牛 NAS?-icode9专业技术文章分享
- 2024-10-03如何安装飞牛 TV 并连接到影视服务器?-icode9专业技术文章分享
- 2024-10-03如何在PVE和ESXI上安装飞牛私有云 fnOS?-icode9专业技术文章分享
- 2024-10-03fnOS国产最强NAS安装系统异常情况处理-icode9专业技术文章分享
- 2024-10-03飞牛NAS如何创建存储空间?-icode9专业技术文章分享
- 2024-10-03fnOS国产最强NAS硬盘会自动休眠吗?-icode9专业技术文章分享
- 2024-10-03fnOS国产最强NAS如何安装飞牛影视和创建媒体库?-icode9专业技术文章分享
- 2024-10-03fnOS国产最强NAS如何为家人朋友开通影视账号?-icode9专业技术文章分享