LeetCode系列记录我学习算法的过程。
携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第15天,点击查看活动详情
题目
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
输入: 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
};
总结一下递归的核心:
1、首先找结束条件 2、只考虑当前该做什么,不考虑下一次该做什么 3、默认递归函数已经实现功能,直接拿来使用 4、考虑每次调用应该返回什么
例如本题:
1、找结束条件,当根节点不存在,或左右子树都不存在时,递归结束
2、如果只有一颗子树存在,那就默认当前递归函数已经实现了获取二叉树最小深度的功能,调用函数传入存在的子树,将返回的结果加上当前深度 1
3、如果两颗树都存在,同理调用函数获取两棵树的最小深度,然后返回其中较小的值 + 1 即可
绝大多数的递归题目都可以用这种方式来解答