257. 二叉树的所有路径

"""
257. 二叉树的所有路径
给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。

示例 1：
输入：root = [1,2,3,null,5]
输出：["1->2->5","1->3"]
示例 2：

输入：root = [1]
输出：["1"]
"""
class TreeNode(object):
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None

class Solution(object):
    def binaryTreePaths(self, root):
        res = []
        def pre_order(root1, str1):
            if not root1: return
            str1 += str(root1.val) + '->'
            if not root1.left and not root1.right:
                res.append(str1.rstrip('->'))

            pre_order(root1.left, str1)
            pre_order(root1.right, str1)
        pre_order(root, '')
        return res

if __name__ == "__main__":
    root = TreeNode(1)
    root.left = TreeNode(2)
    root.right = TreeNode(3)
    res = Solution().binaryTreePaths(root)
    print(res)