获取树节点的子节点计数

# 定义树节点类
class TreeNode:
    def __init__(self, value):
        self.value = value
        self.children = []

# 定义函数获取子节点计数
def get_child_count(node):
    if node is None:
        return 0
    
    child_count = 0
    for child in node.children:
        child_count += get_child_count(child)
    
    return child_count + len(node.children)

# 创建一个示例树
root = TreeNode(1)
child1 = TreeNode(2)
child2 = TreeNode(3)
child3 = TreeNode(4)
child4 = TreeNode(5)

root.children = [child1, child2]
child2.children = [child3, child4]

# 获取根节点的子节点计数
count = get_child_count(root)
print(count)  # 输出结果为3