二叉树——7.二叉树的所有路径

力扣题目链接

给定一个二叉树,返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

输出:["1->2->5","1->3"]

题目要求返回从根节点到叶子节点的路径,所以在遍历方法中选择前序遍历。在具体操作中,先找到最左侧的路径,然后回溯到上一个路径去该节点的右子树(如果没有右子树就再回溯到上一节点),直到遍历完所有路径。

完整代码如下:

# Definition for a binary tree node.
class Solution:
    def binaryTreePaths(self, root):
        result = []
        path = []
        if not root:
            return result
        self.traversal(root, path, result)
        return result

    def traversal(self, cur, path, result):
        path.append(cur.val)  # 中
        if not cur.left and not cur.right:  # 到达叶子节点
            sPath = '->'.join(map(str, path))
            result.append(sPath)
            return
        if cur.left:  # 左
            self.traversal(cur.left, path, result)
            path.pop()  # 回溯
        if cur.right:  # 右
            self.traversal(cur.right, path, result)
            path.pop()  # 回溯

首先,定义binaryTreePaths函数。函数中result为空列表用来存储所有路径,path为空列表用来临时存储当前遍历路径上的节点值。如果根节点不存在,就返回result。接着,调用traversal函数,开始深度优先遍历整棵树。

接着,定义traversal函数。需要输入的包括,当前的节点cur,当前节点所在的路径上的节点值path,以及最终的result。先将当前节点保存在path中。对于节点开始处理,首先是到达了叶子节点(即该节点下没有节点了),根据题目示例中,路径节点之间是由符号连接的,将当前节点路线path用sPath = '->'.join(map(str, path))将path中节点拼接为字符串,用->连接,最后将其加入最终的结果数组result。

开始进行遍历及回溯,如果当前节点存在左子树,就调用traversal继续进行遍历,在递归调用结束后,在path中pop掉(推出)最后一个节点值,即回溯到了上一节点,继续遍历。右子树同理。

总结:

  • 这个算法使用深度优先搜索 (DFS) 方法来遍历二叉树,记录从根节点到每个叶子节点的路径。
  • 每当到达一个叶子节点时,构造一个代表路径的字符串,并将其添加到结果列表中。
  • 通过回溯path.pop()来确保每次递归调用后路径能够恢复到正确的状态,从而继续其他分支的遍历。
  • 8
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值