遍历树并收集节点组合的解决方法可以使用递归或迭代两种方式来实现。以下是两种方法的代码示例:
class TreeNode:
def __init__(self, value):
self.value = value
self.left = None
self.right = None
def collect_nodes_recursive(node, result):
if node is None:
return
result.append(node.value) # 将节点添加到结果集中
# 递归遍历左子树和右子树
collect_nodes_recursive(node.left, result)
collect_nodes_recursive(node.right, result)
# 示例用法
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
result = []
collect_nodes_recursive(root, result)
print(result) # 输出: [1, 2, 4, 5, 3]
class TreeNode:
def __init__(self, value):
self.value = value
self.left = None
self.right = None
def collect_nodes_iterative(root):
if root is None:
return []
stack = [root]
result = []
while stack:
node = stack.pop()
result.append(node.value) # 将节点添加到结果集中
# 先将右子节点入栈,再将左子节点入栈,保证左子节点先被遍历
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return result
# 示例用法
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
result = collect_nodes_iterative(root)
print(result) # 输出: [1, 2, 4, 5, 3]
以上是遍历树并收集节点组合的两种解决方法,分别使用递归和迭代的方式来实现。根据实际情况选择适合的方法进行使用。