【二叉树】求根节点到叶节点数字之和

0x00 题目

给你一个二叉树的根节点 ​​root​​​ 树中每个节点都存放有一个 ​​0​​ 到 ​​9​​ 之间的数字
每条从 ​​根​​ 节点到 ​​叶​​ 节点的路径都代表一个 ​​数字​​

例如，从 ​​根​​​ 节点到 ​​叶​​​ 节点的路径 ​​1 -> 2 -> 3​​​ 表示数字 ​​123​​

计算从根节点到叶节点生成的 ​​所有数字之和​​​ 叶节点 是指 ​​没有​​ 子节点的节点

0x01 思路

根据题目给出的例子：
从 ​​​根​​​ 节点到 ​​叶​​​ 节点的路径 ​​1 -> 2 -> 3​​​ 表示数字 ​​123​​​ 说明每往下走一层，上一层的数字都要乘以 ​​10​​ 一直往下累加，直到叶节点
计算完一条路径后，再计算其他路径
直到计算完所有路径，得出结果

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 sumNumbers(_ root: TreeNode?) -> Int {
    guard let root = root else { return 0 }
    
    // 最终结果
    var res = 0
    
    func sum(_ root: TreeNode?, _val: Int) {
        guard let root = root else { return }
        
        // 上一层的结果 乘以 10，加上本节点的值
        let out = val * 10 + root.val
        // 到了 叶节点
        if root.left == nil && root.right == nil {
            res += out
            return
        }
        
        // 去下一层
        sum(root.left, out)
        sum(root.right, out)
    }
    
    //计算
    sum(root, 0)
    
    return res
}