Skip to main content

二叉树的最小深度

题意

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

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

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

示例 1:

    3
/ \
9 20
/ \
15 7

输入: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

解法

对二叉树进行层序遍历,一层一层遍历,当遇到第一个叶子节点时,就是最小深度。

/**
* 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) {
const stack = [];
if(root == null){
return 0;
}
stack.push(root);
let flag = true; // 标识是否要继续查找
let deepNum = 0;

while(stack.length > 0 && flag){
let size = stack.length;

while(size > 0 && flag){
const current = stack.shift();

if(current.left == null && current.right == null){
// 找到叶子节点
flag = false;
break;
}else{
current.left !== null && stack.push(current.left);
current.right !== null && stack.push(current.right);
}

size--;
}

deepNum++;
}

return deepNum;
};