小编use*_*291的帖子

Python 将列表转换为树表示格式

我和我的朋友正在做一个简单的 Python 项目。实际上,我们正在以自己的方式实现前缀并行求和算法。

我们正在创建和处理一个非常奇怪的格式的二叉树。我们希望将此格式转换为 Tree 打印库/软件(如 ete2)接受的格式。

因此,树的每一层都以这种方式推送到列表中

[ [level0], [level1], ... [level i-1], [root] ]
Run Code Online (Sandbox Code Playgroud)

在我们的格式中,每个内部列表(树的级别)都有偶数个节点或叶子。

例如,假设我们有这个输入:[1, 2, 3, 4, 5]。这将产生以下输出列表:[[1, 2, 3, 4], [3, 7], [10, 5], [15]]

上述输出示例的问题在于,有时叶子不在最后一层,但它们包含在上层列表中。这使得处理列表列表和区分节点和叶子并将它们排列在正确位置变得困难。

我们想将其可视化如下:

http://i.imgur.com/BKrqNZi.png

其中括号中的数字是节点,其他数字是叶子。

为了产生这个输出树,我们想使用一个树绘图库。他们中的大多数人期望这种格式:[root, [left], [right]]

所以,在我们的例子中,我们的格式应该是这样的:

[15, [10, [3, [1], [2]], [7, [3], [4]] ], [5] ]
Run Code Online (Sandbox Code Playgroud)

因为我们目前无法重写代码的整个逻辑,所以我们正在寻找一种巧妙的方法将我们奇怪的格式转换成那种格式。

欢迎任何想法。非常感谢您提前。

python binary-tree list prefix-sum

0
推荐指数
1
解决办法
5881
查看次数

标签 统计

binary-tree ×1

list ×1

prefix-sum ×1

python ×1