6-1 定义一个Tree类
时间: 2023-07-10 13:27:48 浏览: 41
好的,根据您的要求,我为您定义了一个Tree类,具体实现如下所示:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
class Tree:
def __init__(self, root=None):
self.root = root
def insert(self, val):
if not self.root:
self.root = TreeNode(val)
return
node = self.root
while True:
if val < node.val:
if not node.left:
node.left = TreeNode(val)
return
node = node.left
else:
if not node.right:
node.right = TreeNode(val)
return
node = node.right
def inorder(self, node):
if not node:
return
self.inorder(node.left)
print(node.val)
self.inorder(node.right)
```
这里定义了两个类,一个是TreeNode,表示二叉树的一个节点,包含节点的值以及左右子节点;另一个是Tree,表示整个二叉树,包含根节点。在Tree类中,我们可以使用root属性来表示整个二叉树的根节点。在insert方法中,我们可以向二叉树中插入新的节点。在inorder方法中,我们可以对二叉树进行中序遍历,并打印出每个节点的值。这样,就可以实现一个基本的二叉树。