python生成二叉树_python 二叉树

激活谷笔记 • 2025-01-19 08:08 • 阅读 120

在Python中创建二叉树通常涉及定义一个节点类和一个二叉树类。节点类包含节点的值以及指向左右子节点的引用，而二叉树类包含根节点以及用于操作树（如插入、搜索等）的方法。以下是一个简单的示例，展示了如何定义节点类和二叉树类，以及如何创建一个二叉树：

 class TreeNode: def __init__（self, val=0, left=None, right=None）: self.val = val self.left = left self.right = right class BinaryTree: def __init__（self, root=None）: self.root = root def insert（self, value）: if self.root is None: self.root = TreeNode（value） else: self._insert（self.root, value） def _insert（self, node, value）: if value < node.val: if node.left is None: node.left = TreeNode（value） else: self._insert（node.left, value） else: if node.right is None: node.right = TreeNode（value） else: self._insert（node.right, value） def search（self, value）: return self._search（self.root, value） def _search（self, node, value）: if node is None or node.val == value: return node if value < node.val: return self._search（node.left, value） return self._search（node.right, value） 创建二叉树实例 创建空二叉树 tree = BinaryTree（） 插入节点 tree.insert（5） tree.insert（3） tree.insert（7） tree.insert（2） tree.insert（4） tree.insert（6） tree.insert（8） 搜索节点 search_value = 6 result = tree.search（search_value） if result: print（f"找到节点，值为：{result.val}"） else: print（"未找到节点"）

这个例子中，`TreeNode` 类定义了二叉树中的节点，包含节点值 `val` 和指向左右子节点的引用 `left` 和 `right`。`BinaryTree` 类定义了二叉树，包含根节点 `root` 以及插入和搜索节点的方法。

创建二叉树的过程包括：

1. 创建一个空的 `BinaryTree` 实例。

2. 使用 `insert` 方法向树中添加节点。

3. （可选）使用 `search` 方法查找树中的节点。

以上代码展示了如何创建一个简单的二叉搜索树，并执行插入和搜索操作

python生成二叉树_python 二叉树

相关推荐