python binary pythonbinarytree库

转载

mob6454cc7203e2 2023-07-10 20:04:14

文章标签 python binary BinaryTree Python 二叉树 github 文章分类 Python 后端开发

作者：joohwan 翻译：赖信涛责编：仲培艺

python binary pythonbinarytree库_python binary

学过二叉树的朋友都有过这样的经历：按照二叉树的数据手动模拟画出来二叉树。但是现在，有了BinaryTree这个库，你可以不必费这个麻烦了！

BinaryTree是一个小型的Python库，给你提供了简单的API，可以依照树的形式打印一个二叉树，以及二叉树的信息概览。你可以专注于你的算法了！

安装

通过Pypi安装稳定版：

~$ pip install binarytree

从Github安装最新版本：

~$ git clone https://github.com/joowani/binarytree.git
~$ python binarytree/setup.py install

取决于你环境的不同，可能会需要sudo权限。

入门

默认情况下，二叉树使用下面的class作为节点：

class Node(object):

    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None

使用下面的方式以漂亮的形式打印二叉树：

from binarytree import tree, bst, heap, pprint

# Generate a random binary tree and return its root
my_tree = tree(height=5, balanced=False)

# Generate a random BST and return its root
my_bst = bst(height=5)

# Generate a random max heap and return its root
my_heap = heap(height=3, max=True)

# Pretty print the trees in stdout
pprint(my_tree)
pprint(my_bst)
pprint(my_heap)

也支持list形式的二叉树：

from heapq import heapify
from binarytree import tree, convert, pprint

my_list = [7, 3, 2, 6, 9, 4, 1, 5, 8]

# Convert the list into a tree and return its root
my_tree = convert(my_list)

# Convert the list into a heap and return its root
heapify(my_list)
my_tree = convert(my_list)

# Convert the tree back to a list
my_list = convert(my_tree)

# Pretty-printing also works on lists
pprint(my_list)

快速检查二叉树的各个属性：

from binarytree import tree, inspect

my_tree = tree(height=10)

result = inspect(my_tree)
print(result['height'])
print(result['node_count'])
print(result['leaf_count'])
print(result['min_value'])
print(result['max_value'])
print(result['min_leaf_depth'])
print(result['max_leaf_depth'])
print(result['is_bst'])
print(result['is_max_heap'])
print(result['is_min_heap'])
print(result['is_height_balanced'])
print(result['is_weight_balanced'])

导入Node class然后构建你自己的树：

from binarytree import Node, pprint

root = Node(1)
root.left = Node(2)
root.right = Node(3)
root.left.left = Node(4)
root.left.right = Node(5)

pprint(root)

如果默认的Node不能满足你的需要，你可以自定义Node：

from binarytree import Node, setup, tree, pprint

# Define your own null/sentinel value
my_null = -1

# Define your own node class
class MyNode(object):

    def __init__(self, data, left, right):
        self.data = data
        self.l_child = left
        self.r_child = right

# Call setup in the beginning to apply your specification
setup(
    node_init_func=lambda v: MyNode(v, my_null, my_null),
    node_class=MyNode,
    null_value=my_null,
    value_attr='data',
    left_attr='l_child',
    right_attr='r_child'
)
my_custom_tree = tree()
pprint(my_custom_tree)