您当前的位置:首页 > 计算机 > 编程开发 > Python

Python binarytree模块的用法

时间:10-01来源:作者:点击数:

二叉树库需要安装,推荐使用PIP进行安装。安装命令如下:

pip install binarytree

目前的版本是 binarytree-4.0.0。安装后,需要引入该库才可以使用,方法如下:

import binarytree

树的节点使用 binarytree.Node 类来表示。例如,如果希望创建一个值为 10 的树的节点,可以用下面的方法来实现。

>>> root = binarytree.Node(1)
>>> type(root)
<class 'binarytree.Node'>
>>> root.value
1

Node 节点包括以下有用的属性。

  • value:节点的值。
  • right:右孩子节点。
  • left:左孩子节点。

可以通过创建节点并修改节点的 left 和 right 属性来创建一棵树。

>>> root = binarytree.Node(1)
>>> root.left = binarytree.Node(2)
>>> root.right = binarytree.Node(3)
>>> print(root)
  1
/ \
2   3
>>> root.inorder                  # 中序遍历
[Node(2), Node(1), Node(3)]
>>> root.preorder                 # 前序遍历
[Node(1), Node(2), Node(3)]
>>> root.postorder                # 后续遍历
[Node(2), Node(3), Node(1)]
>>> root.height                   # 高度
1
>>> root.size                     # 节点个数
3
>>> root.levels                   # 广度优先遍历
[[Node(1)], [Node(2), Node(3)]]

下面创建一个稍微复杂一点的树。

>>> root = binarytree.Node(1)
>>> root.left = binarytree.Node(2)
>>> root.right = binarytree.Node(3)
>>> root.left.left = binarytree.Node(4)
>>> root.left.right = binarytree.Node(5)
>>> root.levels
[[Node(1)], [Node(2), Node(3)], [Node(4), Node(5)]]
>>> print(root)
    1
  /   \
 2     3
/ \
4  5
>>> root.height
2
>>> root.size
5

 

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门