二叉树建立
时间: 2023-07-24 12:08:22 浏览: 117
二叉树是一种广泛应用于计算机科学中的数据结构,用于存储有层次关系的数据。下面介绍两种建立二叉树的方法。
1. 递归建树
递归建树是一种比较简单的方法,其基本思路是:
- 如果当前节点为空,则创建一个新节点并将数据赋值给它;
- 如果当前节点不为空,则比较待插入数据和当前节点的大小,如果待插入数据比当前节点小,则递归调用左子树,否则递归调用右子树。
下面是递归建树的示例代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def insert(root, val):
if root is None:
return TreeNode(val)
if val < root.val:
root.left = insert(root.left, val)
else:
root.right = insert(root.right, val)
return root
```
2. 迭代建树
迭代建树是一种较为复杂的方法,需要利用栈来辅助建立二叉树。其基本思路是:
- 创建一个空根节点,将待插入数据赋值给根节点;
- 从第二个数据开始遍历,对于每个数据,依次与根节点比较大小,如果比根节点小,则作为左子树的根节点,否则作为右子树的根节点;
- 使用一个栈来保存访问过的节点,每次访问节点时将其入栈;
- 如果当前节点为空,则从栈中弹出一个节点作为当前节点。
下面是迭代建树的示例代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def insert(root, val):
if root is None:
return TreeNode(val)
stack = [root]
while stack:
node = stack.pop()
if val < node.val:
if node.left is None:
node.left = TreeNode(val)
break
else:
stack.append(node.left)
else:
if node.right is None:
node.right = TreeNode(val)
break
else:
stack.append(node.right)
return root
```
以上两种方法都可以用于建立二叉搜索树,其中递归建树是更为常见的方法。
阅读全文
相关推荐


















