111. 二叉树的最小深度简单树DFS

Posted 2022-05-24 pre_eminent

题目：

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明：叶子节点是指没有子节点的节点。

示例 1：

 

输入：root = [3,9,20,null,null,15,7]
输出：2
示例 2：

输入：root = [2,null,3,null,4,null,5,null,6]
输出：5
 

---

提示：

树中节点数的范围在 [0, 10^5] 内
-1000 <= Node.val <= 1000

---

思路：

注意：与104.二叉树的最大深度不同

题目中说的是：最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

注意是叶子节点。

什么是叶子节点，左右孩子都为空的节点才是叶子节点！

---

如下图所示：

当左子树为空时，最小深度不是1，而是右子树的最小深度！

---

 

解答：

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) 
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * 
 */
/**
 * @param TreeNode root
 * @return number
 */
var minDepth = function(root) 
    return dfs(root);
;
function dfs(root) 
    if (!root) 
        return 0;
    
    // 与 求最大深度不同的是：
    // 当左、右子树都存在时，才是如下判断
    if (root.left && root.right) 
        return 1 + Math.min(dfs(root.left), dfs(root.right));
    

    // 如果左子树为空，右子树有值
    // 最小深度不是1，而是 1 + 右子树的最小深度
    if (!root.left && root.right) 
        return 1 + dfs(root.right);
    

    // 如果左子树有值，右子树为空
    // 最小深度不是1，而是 1 + 左子树的最小深度
    if (root.left && !root.right) 
        return 1 + dfs(root.left);
    

    // 当左子树和右子树都为空时
    return 1;