[数据结构与算法]【树-简单】257. 二叉树的所有路径

【题目】
给定一个二叉树，返回所有从根节点到叶子节点的路径。

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

示例:

输入:

1
/
2 3

5

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

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
【代码】
【迭代】

class Solution:
    def binaryTreePaths(self, root: TreeNode) -> List[str]:
        if not root:
            return []
        stack=[]
        ans=[]
        prev=None
        while stack or root:
            while root:
                stack.append(root)
                root=root.left
            root=stack.pop()
            if root.right and prev!=root.right:
                stack.append(root)
                root=root.right
            else:
                if not root.left and not root.right:
                    s=""
                    for x in stack:
                        s+=str(x.val)+"->"
                    s+=str(root.val)
                    ans.append(s)
                prev=root
                root=None
        return ans

【递归】

class Solution:
    def visit(self,root):
        if not root:
            return
        self.ans.append(str(root.val))
        self.visit(root.left)
        self.visit(root.right)
        if not root.left and not root.right:
            self.res.append("->".join(self.ans))
        self.ans.pop()
    def binaryTreePaths(self, root: TreeNode) -> List[str]:
        self.ans=[]
        self.res=[]
        self.visit(root)
        return self.res