首页 > 编程语言 > 详细

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

时间:2020-11-20 17:34:09      阅读:30      评论:0      收藏:0      [点我收藏+]

给定一个仅包含数字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
        }
    }
}

 

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

原文:https://www.cnblogs.com/duzhaoquan/p/14011434.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!