LeetCode 257. 二叉树的所有路径

257. 二叉树的所有路径

Difficulty: 简单

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

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

示例:

输入:

   1
 /   \
2     3
 \
  5

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

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

Solution

Language: 全部题目

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
​
class Solution:
    def binaryTreePaths(self, root: TreeNode) -> List[str]:
        def helper(root):
            if not root: 
                return []
            if not root.left and not root.right:
                return [[root.val]]
            tmp = helper(root.left) + helper(root.right)
            return [[root.val] + i for i in tmp]
        
        res = helper(root)
        return ['->'.join(str(d) for d in i) for i in res]