binaryTree递归循环返回None

时间:2019-04-26 01:44:13

标签: python binary-tree

我有一个称为print_tree_array的函数,该函数应该从左到右打印出一个二叉树的数组,例如:[460, 517, 72, 33, 699, 789, 764, 685, 961, 851]。这样做很好,直到最后一个循环。然后,即使我在None语句上执行return array,也只会返回else

我尝试了很多事情,并得出结论,二叉树有效且“框架”已正确设置。

class node:
    def __init__(self, value=None):
        self.value= value
        self.r_child=None
        self.l_child=None

class tree:
    def __init__(self):
        self.root=None

    def insert(self, value, cur_node):
        if(self.root==None):
            self.root=node(value)
        else:
            if(value<cur_node.value):
                if(cur_node.l_child==None):
                    cur_node.l_child=node(value)
                else:
                    self.insert(value, cur_node.l_child)    
            elif(value>cur_node.value):
                if(cur_node.r_child==None):
                    cur_node.r_child=node(value)
                else:   
                    self.insert(value, cur_node.r_child)    

    def print_tree_array(self, cur_node, array=[]):
        if(self.root==None):
            return []
        else:
            if(cur_node!=None):
                self.print_tree_array(cur_node.l_child, array)
                self.print_tree_array(cur_node.r_child, array)
                print(array)
                return array.append(cur_node.value)
            else:
                return array        

    def get_root(self):
        return self.root 


def fill_with_random(tree, size, min_range, max_range):
    from random import randint
    for _ in range(size):
        value = randint(min_range, max_range)
        print(value)
        tree.insert(value, tree.root)
    return tree

current_tree = tree()
current_tree = fill_with_random(tree=current_tree, size=10, min_range=0, max_range=999)
print(current_tree.print_tree_array(cur_node=current_tree.root))
708
999
534
79
692
39
904
979
46
568
[]
[46]
[46, 39]
[46, 39, 79]
[46, 39, 79, 568]
[46, 39, 79, 568, 692]
[46, 39, 79, 568, 692, 534]
[46, 39, 79, 568, 692, 534, 979]
[46, 39, 79, 568, 692, 534, 979, 904]
[46, 39, 79, 568, 692, 534, 979, 904, 999]
None

这就是它输出的内容,您可以看到它使用第一行打印的行填充了二叉树。然后,它继续将树的每个“端”附加到数组中。但是它无法追加最右边的项目并返回整个数组。

1 个答案:

答案 0 :(得分:0)

array.append实际上不返回任何内容,而只是追加到原始列表中

https://www.programiz.com/python-programming/methods/list/append

如果需要,应在附加后返回数组。