给定一棵二叉树，该文章详细介绍了如何使用递归和迭代两种方法来找到从根节点到叶子节点的所有路径。在递归方法中，如果节点没有左右子树，则返回其值；否则，将路径与子树路径组合。在迭代方法中，利用栈存储节点和路径，遍历每个节点直到找到叶子节点。示例输入和输出也一同给出，帮助理解算法的工作原理。
给定一颗二叉树，返回它从根结点到叶子结点的所有路径
注意：
叶子结点没有子树
Example:
Input:
1
/ \
2 3
\
5
Output: ["1->2->5", "1->3"]
Explanation: All root-to-leaf paths are: 1->2->5, 1->3
1：递归
def binaryTreePaths(self, root):
"""
:type root: TreeNode
:rtype: List[str]
"""
if not root:
return []
if not root.left and not root.right:
return [str(root.val)]
pathList = []
if root.left:
pathList += self.binaryTreePaths(root.left)
if root.right:
pathList += self.binaryTreePaths(root.right)
for index, path in enumerate(pathList):
pathList[index] = str(root.val) + "->" +path
return pathList
2：迭代(参考他人)
def binaryTreePaths(self, root):
"""
:type root: TreeNode
:rtype: List[str]
"""
if not root:
return []
pathList, stack = [], [(root, "")]
while stack:
node, pathStr = stack.pop(0)
if not node.left and not node.right:
pathList.append(pathStr + str(node.val))
if node.left:
stack.append((node.left, pathStr + str(node.val) + "->"))
if node.right:
stack.append((node.right, pathStr + str(node.val) + "->"))
return pathList
算法题来自：https://leetcode-cn.com/problems/valid-anagram/description/
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
