(Ipython)Matplotlib 中将二叉树可视化

Vogelbaum

已于 2022-03-02 23:46:29 修改

阅读量1.8k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：数据结构算法文章标签：算法二叉树数据可视化

于 2022-03-01 07:39:58 首次发布

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/ChillingKangaroo/article/details/123196629

本文介绍如何利用IPython和Matplotlib将二叉树进行可视化，特别是在学习和调试二叉树操作时能更直观地发现问题。通过创建二维矩阵并计算节点的偏移量，将二叉树结构绘制出来。虽然算法未做优化，对于较高层次的树可能会较慢，但适合初学者理解二叉树结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

（注意之前代码有错误目前已更新）

最近学习黑红二叉树，我想如果把二叉树可视化在操作的时候如果出错会比较容易发现。

在网上搜了一圈只有比较简单的ascii 的代码。

自己用Ipython写了一个，比较适合学生。

PS：算法没有做优化，加上matplotlib本身就慢，不适合较高的树。

效果见图：

基本算法：

首先获取二叉树的高度 h，二叉树在所有节点都有左右在子节点的情况下，在高度 y时 X轴包含节点 $2^{(h-1)}$ 个。
比如高度 2 有 $2^{(2-1)}$ = 2个节点.
创建一个Y轴长度相同的二维矩阵，X轴包含了所有节点和节点左右的空白区域，长度为 $2^{(h-1) }* 2 +1$ 。
在遍历矩阵中所有元素的时候，如果坐标 (x,y)中包含了一个节点则检测它的子节点，这时候把它的子节点放入到二维矩阵当中
如左侧子节点坐标为 (x_,y_),

最低0.47元/天解锁文章

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。