要求:求二叉树最小深度
算法:DFS深度优先搜索算法
几种特殊的情况
空树
判断的当前节点为空,则需要返回最小深度为0
只有一个节点
这一个节点就是叶子结点,无需递归,但深度也需要加一
两个结点
根节点的左节点递归一次,深度为2,右节点如果直接递归,递归后会判断当前节点为空,会把最小深度置为0,所以需要判断:当子节点不为空的时候才递归搜索
深度很大
当深度很大的时候,最小深度的值会小于实际数的最小深度,导致最小深度值不变,所以需要为最小深度的初始值设置一个最大的值
代码测试
package com.robot.algorithm;
/**
* 求二叉树最小深度。
*
* @author 张宝旭
*/
public class BinaryThreeMinDepth {
// 先令最小深度为最大值
int minDep = Integer.MAX_VALUE;
public int minDepth(TreeNode root) {
dfs(root, 0);
return minDep;
}
// dfs深度优先搜索
// 不是叶子节点时,深度加一进行递归,递归到叶子节点,深度再加一
public void dfs(TreeNode node, int dep) {
// 如果节点为空,说明是一颗空树
if (node == null) {
minDep = 0;
return;
}
// 如果是叶子结点,深度加一,然后返回
if (node.left == null && node.right == null) {
dep++;
// 更新最小深度
if (dep < minDep) {
minDep = dep;
}
return;
}
// 如果左节点不为空,则递归左节点
if (node.left != null) {
dfs(node.left, dep + 1);
}
// 如果右节点不为空,则递归右节点
if (node.right != null) {
dfs(node.right, dep + 1);
}
}
}
图示,此二叉树的最小深度为2