Question

我必须递归计算二叉树中的节点。我是python的新手，无法为我的问题找到任何解决方案来完成代码。

这是我已经尝试过的。如您所见，它还不完整，我不知道该去哪里。

class Tree:
    def __init__(self, root):
        self.root = root

    def add(self, subtree):
        self.root.children.append(subtree)

class Node:
    def __init__(self, value, children=None):
        self.value = value
        self.children = children if children is not None else []

def check_root(tree):
    if tree.root is None:
        return 0
    if tree.root is not None:
        return count_nodes(tree)

def count_nodes(tree):
    if tree.root.children is not None:
        j = 0
        for i in tree.root.children:
            j = 1 + count_nodes(tree)
        return j


print(count_nodes(Tree(None))) # 0
print(count_nodes(Tree(Node(10)))) # 1
print(count_nodes(Tree(Node(5, [Node(6), Node(17)])))) #3

每执行一个新步骤，我都会得到不同的错误。例如。使用此代码，我已经超过了最大递归深度。

感谢您的阅读。任何提示或帮助下一步做什么将不胜感激。

Answer 1

一种简便方法：

让我们假设， A 是一个二叉树，其子节点或节点不是 NULL 。例如

现在，为了计算节点数，我们有一个简单的解决方法。

递归方法：>>> get_count（root）

对于二叉树，递归的基本思想是在 Post-Order 中遍历该树。在这里，如果当前节点已满，我们将结果增加 1并增加 1 ，并添加左右子树的返回值，例如：

class TestNode(): 


def __init__(self, data):  
    self.data = data 
    self.left = None
    self.right = None

现在，我们继续使用以下方法获取二叉树中的完整节点数：

def get_count(root): 

  if (root == None): 
    return 0

  res = 0
  if (root.left and root.right): 
     res += 1

  res += (get_count(root.left) + 
        get_count(root.right))  
  return res

最后，为了运行代码，我们将管理一个主作用域：在这里，我们创建了上面给出的二叉树A ：

if __name__ == '__main__': 

  root = TestNode(3)  
  root.left = TestNode(7)  
  root.right = TestNode(5)  
  root.left.right = TestNode(6)  
  root.left.right.left = TestNode(1)  
  root.left.right.right = TestNode(4)

现在最后，在主作用域内，我们将打印二叉树节点的数量，例如：

  print(get_Count(root))

这是递归函数获取二叉树 A 的get_count的时间复杂度。

时间复杂度： O（n）

Answer 2

我首先将根节点传递给count_nodes函数-

print(count_nodes(Tree(None)).root) # 0
print(count_nodes(Tree(Node(10))).root) # 1
print(count_nodes(Tree(Node(5, [Node(6), Node(17)]))).root) #3

或为此提供帮助功能。

然后count_nodes函数可以看起来像这样

def count_nodes(node):
    return 1 + sum(count_nodes(child) for child in node.children)

编辑：我刚刚注意到，您可以拥有一个None根，这意味着，您还应该处理该问题：

def count_nodes(node):
    if node is None:
        return 0
    return 1 + sum(count_nodes(child) for child in node.children)

如果您真的要在一个函数中处理树或节点，则可以使其更加难看：

def count_nodes(tree_or_node):
    if isinstance(tree_or_node, Tree):
        return count_nodes(tree_or_node.root)
    if tree_or_node is None:
        return 0
    return 1 + sum(count_nodes(child) for child in tree_or_node.children)

然后您可以像最初一样调用它。

Answer 3

您的问题是您要无限计数同一棵树。看看这一行：

j = 1 + count_nodes(tree)

递归计数二叉树中的节点

3 个答案: