class TreeNode: def __init__(self, value): self.value = value self.left = None self.right = None def add_node(self, value): if self.root is None: self.root = TreeNode(value) else: self._add_node_recursive(self.root, value) def _add_node_recursive(self, node, value): if value < node.value: if node.left is None: node.left = TreeNode(value) else: self._add_node_recursive(node.left, value) else: if node.right is None: node.right = TreeNode(value) else: self._add_node_recursive(node.right, value) def height(self,node): """Returns the height of the longest branch of the tree.""" def height_rec(t): if not t: return 0 else: return max(1+height_rec(t.left), 1+height_rec(t.right)) return height_rec(node)
时间: 2024-02-14 19:20:31 浏览: 98
jqueryeasyui_tree_comtree.docx
这段代码实现了二叉树的添加节点和获取二叉树高度的方法。下面是代码的详细解释:
首先定义了 `TreeNode` 类表示二叉树的节点,包含节点的值以及左右子节点。然后定义了 `add_node` 方法用于添加节点到二叉树中。如果树为空,则创建一个新节点作为根节点,否则调用 `_add_node_recursive` 方法将节点添加到正确的位置。
`_add_node_recursive` 方法使用递归实现,比较新节点的值和当前节点的值,如果小于当前节点的值,则递归处理左子树,否则递归处理右子树。如果左子树或右子树为空,则将新节点添加到该位置,否则继续递归处理。
最后定义了 `height` 方法用于获取二叉树的高度,使用了内部的 `height_rec` 方法来递归计算节点的高度。如果节点为空,则返回0,否则分别计算左子树和右子树的高度,并取其中最大值再加上1就是当前节点的高度。
阅读全文