题解 | #二叉树的最小深度#

package main
import "math"
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return int整型
*/
func run(root *TreeNode) int {
	// write code here
    if root == nil {
        return 0
    }

    l := run(root.Left)    // 递归计算出左子树的高度
    r := run(root.Right)   // 递归计算出右子树的高度
    if l == 0 && r != 0 {        // 如果左子树为空而右子树不为空，则直接返回右子树高度+1
        return r + 1
    } else if r == 0 && l != 0 { // 与上同理，返回左子树高度+1
        return l + 1
    } else if l != 0 && r != 0 {    // 如果左右子树都存在，则返回较小的子树高度+1
        return int(math.Min(float64(l), float64(r))) + 1
    } else {
        return 1                // 如果左右子树都为空，则直接返回1
    }
}