非二叉树递归

时间:2017-12-19 21:32:41

标签: python recursion tree

我正在尝试创建一个程序来设置一个非二叉树,每个节点都连接到子节点。在这个测试示例中,为简单起见,我使用了二叉树。输入是:

1
3   5
4   6

(数字之间使用制表符)。 我试图从根(1)开始生成树,其子节点为3和5,并且每个节点都有子节点4和6。 树形图可能如下所示:

    4
   /
  3
 / \
1   6
 \ /
  5
   \
    4

当我尝试将子项添加到树中时,它会创建一个无限循环来调用我的递归函数。我已经把问题缩小到它调用函数,其中分支是1循环,但这里是代码:

# input is a list of branch values
file = open("treehash.txt","r")
input = file.readlines()
for line in range(len(input)):
input[line] = input[line].rstrip().split('\t')
file.close()

# create the tree node
class Tree(object):
    value = None
    children = []
    def __init__(self, value):
        self.value = value

# adds all children to a given parent
def set_children(parent, branch):
    if branch < len(input) - 1:
        for num in input[branch + 1]:
            parent.children.append(Tree(int(num)))
        for child in parent.children:
            set_children(child, branch + 1)

# store all roots in array
roots = []
for root in range(len(input[0])):
    roots.append(Tree(int(input[0][root])))
    set_children(roots[root], 0)

1 个答案:

答案 0 :(得分:1)

如果你像在

中那样在类中编写变量
class Tree(object):
    value = None
    children = []

它们绑定到类,而不是实例。对于value,您使用__init__构造函数中的实例绑定变量覆盖它,但children引用的列表由所有Tree个实例共享。

删除上述变量设置并改为使用:

class Tree(object):
    def __init__(self, value):
        self.value = value
        self.children = []