力扣题目链接
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
输出:["1->2->5","1->3"]
题目要求返回从根节点到叶子节点的路径,所以在遍历方法中选择前序遍历。在具体操作中,先找到最左侧的路径,然后回溯到上一个路径去该节点的右子树(如果没有右子树就再回溯到上一节点),直到遍历完所有路径。
完整代码如下:
# Definition for a binary tree node.
class Solution:def binaryTreePaths(self, root):result = []path = []if not root:return resultself.traversal(root, path, result)return resultdef 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)returnif 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()来确保每次递归调用后路径能够恢复到正确的状态,从而继续其他分支的遍历。