层序遍历的巧用

需求

给定一个仅包含数字0-9 0−9 的二叉树，每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1→2→3,那么这条路径就用123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和

这颗二叉树一共有两条路径，
根节点到叶子节点的路径 12
根节点到叶子节点的路径 13
答案为 12+13=25

常规思路

• 先求得根节点到各自叶子节点路径
• 将路径信息转为数字信息相加求和

golang实现

import "math"
func sumNumbers( root *TreeNode ) int {
    var sum int
    var convert func([]int)int
    var root2leaf func(*TreeNode, []int)
    
    convert = func(path []int) int{
        rs :=0
        for i:=0;i<len(path);i++{
            rs +=path[i]* int(math.Pow10(len(path)-1-i))
        }
        return rs
    }
    root2leaf = func(root *TreeNode,path []int){
        if root == nil {
            return 
        }
        if root.Left==nil && root.Right==nil {
            path = append(path,root.Val)
            sum +=convert(path)
            return 
        }
        root2leaf(root.Left,append(path,root.Val))
        root2leaf(root.Right,append(path,root.Val))
    }
    root2leaf(root,[]int{})
    return sum
}

层序思维

• 充分复用入参二叉树本身层序性
• 先序遍历的思想(根左右)+数字求和
• 每一层都比上层和*10+当前根节点的值

python解法

        def preOrderSum(root,curSum):
            if root is None:
                return 0
            curSum = curSum*10+root.val
            if root.left is None and root.right is None:
                return curSum
            return preOrderSum(root.left, curSum)+preOrderSum(root.right, curSum)
        return preOrderSum(root, 0)

小结

巧思路可以省不少事