怎么用pyhon随机创造一个二叉树
时间: 2024-05-16 16:12:01 浏览: 66
可以使用Python中的random模块来随机创造一个二叉树。首先,定义一个TreeNode类,其中包含了节点的值以及左右子节点。然后,使用递归的方式来创建这个二叉树,每次随机生成一个节点值,将它作为当前节点的值,并在左右子节点中随机选择一个节点作为下一个节点。以下是一个示例代码:
```python
import random
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def create_random_tree(depth=3):
if depth == 0 or random.random() < 0.5:
return None
root = TreeNode(random.randint(0, 100))
root.left = create_random_tree(depth - 1)
root.right = create_random_tree(depth - 1)
return root
# 测试代码
root = create_random_tree()
print(root.val)
print(root.left.val)
print(root.right.val)
```
这段代码会生成一个深度为3的随机二叉树,并打印出根节点以及它的左右子节点的值。你可以自定义深度等参数。
阅读全文