python实现创建二叉树,要求函数输入为一维数组,输出为根节点和二叉树
时间: 2024-01-23 18:46:01 浏览: 96
二叉树的层次遍历实现.docx
5星 · 资源好评率100%
的形式。
好的,这是一个计算机相关的问题,我可以回答。以下是Python代码实现创建二叉树:
```python
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
def create_tree(nums):
if not nums:
return None
mid = len(nums) // 2 # 中间节点为根节点
root = TreeNode(nums[mid])
root.left = create_tree(nums[:mid])
root.right = create_tree(nums[mid+1:])
return root
```
输入为一维数组,如 `[1, 2, 3, 4, 5, 6, 7]`,输出为根节点和二叉树的形式。具体使用方法可以参考下面的代码:
```python
nums = [1, 2, 3, 4, 5, 6, 7]
root = create_tree(nums)
```
这样就可以得到一个二叉树的根节点 `root`。
阅读全文