使用 C++ 实现二叉树搜索功能。二叉搜索树是一种常用的数据结构,适用于各种查找和排序场景。通过递归的方式,我们可以在二叉树中快速搜索特定的值。
函数解释:
-
insert
: 这个函数用于向二叉树中插入一个新节点。首先,它检查根节点是否为空,如果为空,则创建一个新节点,并将其作为根节点。如果根节点不为空,则根据节点值的大小,递归地将新节点插入到左子树或右子树中。最后,返回更新后的根节点。 -
search
: 这个函数用于在二叉树中搜索一个特定的值。首先,它检查根节点是否为空。如果根节点为空,则表示树中不存在目标值,返回 false。如果目标值与根节点的值相等,则表示找到目标值,返回 true。如果目标值小于根节点的值,则递归地在左子树中搜索目标值。如果目标值大于根节点的值,则递归地在右子树中搜索目标值。 -
inorderTraversal
: 这个函数用于对二叉树进行中序遍历,并输出每个节点的值。中序遍历是一种深度优先遍历方式,它首先遍历左子树,然后输出根节点的值,最后遍历右子树。在这个函数中,首先检查根节点是否为空。如果不为空,则先递归地遍历左子树,然后输出根节点的值,最后递归地遍历右子树。 -
main
: 这是程序的主函数。在这个函数中,我们创建一个空的二叉树,并使用insert
函数插入一些节点。然后,我们使用inorderTraversal
inorderTraversal函数对树进行中序遍历,并输出树中的节点。接下来,我们使用
search` 函数搜索树中的两个值:4 和 9。根据搜索结果,我们输出相应的信息。 -
#include <iostream> struct TreeNode { int value; TreeNode* left; TreeNode* right; TreeNode(int val) : value(val), left(nullptr), right(nullptr) {} }; TreeNode* insert(TreeNode* root, int value) { if (root == nullptr) { root = new TreeNode(value); } else if (value < root->value) { root->left = insert(root->left, value); } else { root->right = insert(root->right, value); } return root; } bool search(TreeNode* root, int value) { if (root == nullptr) { return false; } else if (value == root->value) { return true; } else if (value < root->value) { return search(root->left, value); } else { return search(root->right, value); } } void inorderTraversal(TreeNode* root) { if (root == nullptr) { return; } inorderTraversal(root->left); std::cout << root->value << " "; inorderTraversal(root->right); } int main() { TreeNode* root = nullptr; // 插入节点 root = insert(root, 5); root = insert(root, 3); root = insert(root, 7); root = insert(root, 1); root = insert(root, 4); root = insert(root, 6); root = insert(root, 8); // 中序遍历并输出树中的节点 std::cout << "二叉树中的节点(中序遍历):"; inorderTraversal(root); std::cout << std::endl; // 搜索值为 4 的节点 int searchValue = 4; if (search(root, searchValue)) { std::cout << "找到值为 " << searchValue << " 的节点" << std::endl; } else { std::cout << "未找到值为 " << searchValue << " 的节点" << std::endl; } // 搜索值为 9 的节点 searchValue = 9; if (search(root, searchValue)) { std::cout << "找到值为 " << searchValue << " 的节点" << std::endl; } else { std::cout << "未找到值为 " << searchValue << " 的节点" << std::endl; } return 0; }