LeetCode110、二叉树的最小深度

80 阅读2分钟

LeetCode 系列记录我学习算法的过程。

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第15天,点击查看活动详情

题目

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

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

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

示例:

image.png

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

提示:

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

思路

这个题目和求二叉树的最大深度看起来很相似,但是思路却没能一下子就想出来

写了好几遍才找到规律,一样还是用递归的方法来求解:

  • 首先判断根节点是否存在,不存在直接返回 0
  • 再判断左右子树均不存在的情况,返回 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) {
    // 根节点不存在,则返回 0
    if(!root) return 0
    // 两颗子树不存在,返回 1
    if(!root.left && !root.right) return 1

    // 只有左子树,返回递归左子树后的结果 +1
    if (root.left && !root.right) {
        return minDepth(root.left) + 1
    }
    
    // 只有右子树,返回递归右子树后的结果 +1
    if (!root.left && root.right) {
        return minDepth(root.right) + 1
    }

    // 左右子树都存在,分别递归获取结果
    let left = minDepth(root.left)
    let right = minDepth(root.right)

    // 返回较小值 +1
    return Math.min(left, right) + 1
};

image.png

总结一下递归的核心:

1、首先找结束条件 2、只考虑当前该做什么,不考虑下一次该做什么 3、默认递归函数已经实现功能,直接拿来使用 4、考虑每次调用应该返回什么

例如本题:

1、找结束条件,当根节点不存在,或左右子树都不存在时,递归结束

2、如果只有一颗子树存在,那就默认当前递归函数已经实现了获取二叉树最小深度的功能,调用函数传入存在的子树,将返回的结果加上当前深度 1

3、如果两颗树都存在,同理调用函数获取两棵树的最小深度,然后返回其中较小的值 + 1 即可

绝大多数的递归题目都可以用这种方式来解答