257. 二叉树的所有路径

leecode原题

题目

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

叶子节点 是指没有子节点的节点。

示例

示例 1:

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

示例 2:

输入：root = [1]
输出：["1"]

提示：

• 树中节点的数目在范围 [1, 100] 内
• -100 <= Node.val <= 100

解题思路

思路

仔细想一想就是前序遍历的变种，在前序递归的过程中，记录当前递归到的路径，并传到到下一个递归中，遇到叶子节点的时候，才将最终的路径追加到数组中。

实现

源码

递归法

func travel(root *TreeNode, res *[]string, s string) {
    // 如果是叶子结果, 添加路径到数组中
    if root.Left == nil && root.Right == nil {
        *res = append(*res, s+strconv.Itoa(root.Val))
        return
    }
    // 其实就是前序遍历的变种
    s = s + strconv.Itoa(root.Val) + "->"
    if root.Left != nil {
        travel(root.Left, res, s)
    }
    if root.Right != nil {
        travel(root.Right, res, s)
    }
}

// 递归法实现
func binaryTreePaths(root *TreeNode) []string {
    if root == nil {
        return nil
    }
    res := make([]string, 0)
    travel(root, &res, "")
    return res
}