swift算法题:二叉树根节点到叶子节点的所有路径和

给定一个仅包含数字0−9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。

例如根节点到叶子节点的一条路径是1→2→3,那么这条路径就用123 来代替。找出根节点到叶子节点的所有路径表示的数字之和

例如:     1

        2  3

这颗二叉树一共有两条路径,根节点到叶子节点的路径1→2 用数字12 代替,根节点到叶子节点的路径1→3 用数字13 代替,所以答案是12+13=25

//到某一个叶子结点的路径只有一个,那么也就是所有叶子结点的路径之和
func sumNumber(rootNode:TreeNode<Int>) -> Int{
    let temp :Int = 0
    var sum = 0
    leafnum(node: rootNode, path_num: temp, sum: &sum)
    
    return sum
    
}
func leafnum(node:TreeNode<Int>?,path_num:Int,sum:inout Int){
    var temp = 0
    if let root = node {
        temp = path_num * 10 + root.value
        leafnum(node: root.leftNode, path_num: temp, sum: &sum)
        leafnum(node: root.rightNode, path_num: temp, sum: &sum)
        if root.leftNode == nil && root.rightNode == nil{
            sum = sum + temp
        }
    }
}

 

posted @ 2020-11-20 16:07  不停奔跑的蜗牛  阅读(235)  评论(0)    收藏  举报