二叉树二叉树的所有路径

0x00 题目

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

0x01 思路

这个问题就像是：
从山顶往山脚走，有很多条路
你需要把所有下山的路都找出来

既然是找线路，就应该习惯性地想到，使用 深度优先 的遍历方式：递归 ！

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法一：
可以先把一条路径的所有节点，放在数组内，然后生成路径

func binaryTreePaths(_ root: TreeNode?) -> [String] {
    guard let root = root else { return [] }
    
    // 存放所有路径的数组
    var res: [String] = []
    // 存放当前路径的数组
    var tmp: [String] = []
    
    // 递归函数
    func build(_ root: TreeNode?) {
        // 没路了，返回
        if root == nil {
            return
        }
        
        // 添加节点
        tmp.append("\(root!.val)")
        // 往下走
        build(root?.left)
        build(root?.right)
        
        // 叶子节点，走到底了
        if root?.left == nil && root?.right == nil {
            // 存放结果
            res.append(tmp.joined(separator:"->"))
        }
        
        // 移除最后一步，回头找另外的路线
        tmp.removeLast()
    }
    
    // 开始下山啦~
    build(root)
    
    return res
}

解法二：
递归的同时生成路径

func binaryTreePaths(_ root: TreeNode?) -> [String] {
    guard let root = root else { return [] }

    var res: [String] = []
    func build(_ root: TreeNode?, _ path: String) {
        // 叶子节点，走到底了
        if root?.left == nil && root?.right == nil {
            // 存放结果
            res.append(path)
            return
        }
        
        // 左边有路，往左边走
        if let left = root?.left {
            build(left, path   "->\(left.val)")
        }
        // 右边有路，往右边走
        if let right = root?.right {
            build(right, path   "->\(right.val)")
        }
    }
    
    // 开始下山啦~
    build(root, "\(root.val)")
    
    return res
}

小笔记

笔记从未如此简单

这篇好文章是转载于：学新通技术网