257. 二叉树的所有路径

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

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

示例:

输入:

   1
 /   \
2     3
 \
  5

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

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

思路:

  • 二叉树的先序遍历
  • 用一个数组存储所有路径,用helper函数递归遍历二叉树,当遍历到叶子节点时,将路径放入数组中
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def binaryTreePaths(self, root):
        """
        :type root: TreeNode
        :rtype: List[str]
        """
        result = []
        path = ''
        self.helper(root, path, result)
    
        return result

    def helper(self, root, path, result):
        if not root:
            return
        
        path += str(root.val)

        if root.left:
            self.helper(root.left, path+'->', result)
        
        if root.right:
            self.helper(root.right, path+'->', result)

        if not root.left and not root.right:
            result.append(path)
posted @ 2018-09-21 14:31  yuyin  阅读(119)  评论(0编辑  收藏  举报