【leetcode】111.二叉树的最小深度

做这道题目前，可以先做一下【leetcode】104.二叉树的最大深度

题目

leetcode原题链接

给定一个二叉树，找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
**说明：**叶子节点是指没有子节点的节点。

示例 1：

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

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

提示：
树中节点数的范围在 [0, 105] 内
-1000 <= Node.val <= 1000

思路

首先要注意以下最小深度的定义：从根节点到最近叶子节点的最短路径上的节点数量。

所以像下面这棵树的深度应该为3而不是1

递归法

• 我们定义一个递归函数用于求树的最小高度，因为最小高度等于最小深度

• 递归函数的参数和返回值：参数就是要求的树的根节点，返回值就是该树的最小高度。

• 递归结束条件：当遇到空节点时，返回高度为0，即空节点看作高度为0的节点。

• 单层递归的逻辑：这里是容易出错的地方，因为树的最小深度是根节点到最近叶子节点的节点数量，所以要分4种情况讨论：

  • 第一种：左子树为空，右子树不为空，深度为1 + minDepth(node.right)
  • 第二种：左子树不为空，右子树为空，深度为1 + minDepth(node.left)
  • 第三种：左右子树都不为空，深度为1 + Math.min(minDepth(node.right),minDepth(node.left))
  • 第四种：左右子树都为空，深度为1

  其中第三种和第四种可以合并为一种 ，深度为1 + Math.min(minDepth(node.right),minDepth(node.left))

代码

/**
 * 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) {
    if(!root) return 0
    // 计算左右子树最小深度
    const leftDepth = minDepth(root.left)
    const rightDepth = minDepth(root.right)

    if(!leftDepth && rightDepth) return 1 + rightDepth
    if(leftDepth && !rightDepth) return 1 + leftDepth
    return 1 + Math.min(leftDepth , rightDepth)
};

关注我的专栏，每天更新三道leetcode题解，一起变强！