繁体   English   中英

反向打印树级别

[英]Printing tree levels in reverse

我有一个代码在其中打印霍夫曼树。 这部分是:

while len(numArr) > 1:
    numArr = [numArr[0] + numArr[1]] + numArr[2:]
    numArr = sorted(numArr)
    valHold = numArr[0] * 8
    print(numArr)

不用介意valHold变量,我用它来计算输入字符串的未压缩位。

假设我有1,1,1,2,3,4作为列表numArr的元素(这些元素来自Counter并转移到letter_ar r和numArr以便将两者分​​开)。

我只能这样打印:

1,1,1,1,2,3,4
1,1,2,2,3,4
2,2,2,3,4
2,3,4,4
4,4,5
5,8
13

有没有其他方法可以打印? 这样看起来会更像一棵树吗? 像这样:

13
5,8
4,4,5
2,3,4,4
2,2,2,3,4
1,1,2,2,3,4
1,1,1,1,2,3,4

如果您可以教我如何使用缩进来打印,那就更好了:

     13
     5,8
    4,4,5
   2,3,4,4
  2,2,2,3,4
 1,1,2,2,3,4
1,1,1,1,2,3,4

请注意, numArr列表的元素未预定义。 它基于用户在程序中输入的内容。

当然:

tree = []
while len(numArr) > 1:
    numArr = [numArr[0] + numArr[1]] + numArr[2:]
    numArr = sorted(numArr)
    valHold = numArr[0] * 8
    tree.append(numArr)

indent = len(tree)
for row in tree[::-1]:
    print(" " * indent, row)
    indent -= 1

您可以按以下树状格式输出数据:

numArray = [
    [1, 2, 1, 4, 1, 1, 3],
    [2, 4, 1, 3, 2, 1],
    [2, 3, 2, 4, 2],
    [4, 2, 3, 4],
    [5, 4, 4],
    [8, 5],
    [13]]

output = [','.join(str(x) for x in sorted(row)) for row in numArray[::-1]]

for row in output:
    print row.center(len(output[-1]))

这将显示:

      13     
     5,8     
    4,4,5    
   2,3,4,4   
  2,2,2,3,4  
 1,1,2,2,3,4 
1,1,1,1,2,3,4

[::-1]可用于以相反的顺序读取数组。 因此,这里的想法是读取每一行并将每个元素转换为字符串。 然后使用逗号将它们连接起来以创建数字列表。 最后,然后根据最长条目的长度将每行居中显示。

为了以相反的顺序打印,您可以先将其放在列表中,然后再反转。

array = []
while len(numArr) > 1:
    numArr = [numArr[0] + numArr[1]] + numArr[2:]
    numArr = sorted(numArr)
    array.append(numArr)
array.reverse()

要使输出缩进并使其与每个数字对齐可能需要一些额外的工作,但是您可以尝试将输出居中作为一种解决方法。 首先将每个列表转换为字符串,然后计算最大宽度。 然后使用str.center将文本居中。

array_str = list(map(lambda level: ','.join(str(i) for i in level), array))
width = max(len(s) for s in array_str)
for s in array_str:
    print(s.center(width))

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM