遍历树,收集节点组合
创始人
2024-12-04 08:30:33
0

遍历树并收集节点组合的解决方法可以使用递归或迭代两种方式来实现。以下是两种方法的代码示例:

  1. 递归方法 递归方法通过深度优先遍历树的方式,遍历每个节点,并将节点添加到结果集中。递归方法的实现如下:
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]
  1. 迭代方法 迭代方法使用栈来模拟递归的过程,遍历每个节点并将节点添加到结果集中。迭代方法的实现如下:
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]

以上是遍历树并收集节点组合的两种解决方法,分别使用递归和迭代的方式来实现。根据实际情况选择适合的方法进行使用。

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...