在Python中使用GraphViz绘制二叉排序树

最新推荐文章于 2024-10-22 00:17:38 发布

原创

最新推荐文章于 2024-10-22 00:17:38 发布 · 783 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#graphviz #python #二叉树

在Python中使用GraphViz绘制二叉排序树
本文章结合了py和Graphviz，想要快速可视化二叉排序树和想要初步了解Graphviz的小伙伴可以看看。
准备工作：
1.在GraphViz官网下载文件后，在电脑安装，并将安装路径导入环境变量PATH
2.进入cmd，输入 pip install graphviz

实现代码

# python 实现二叉树
# author E
import queue
from graphviz import Digraph

class Node(object):
	def __init__(self, data = None, parent = None):
		self.data = data
		self.parent = parent
		self.right = None
		self.left = None

class BinaryTree(object):
	def __init__(self, list):
		self.root = Node()
		for ele in list:
			self.insert(ele, self.root)

	def insert(self, x, r):
		if r.data == None:
			self.root.data = x
		if x < r.data:
			if r.left == None:
				r.left = Node(x, r