二叉树最小深度问题:Leetcode111

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

最小深度的定义是:从根节点到叶子结点的最短路径上的节点个数。

思路1:递归方式解决最小深度

如图,以值为5的节点为例,如果要找递推公式,我们想到的最简单最初始的递推公式就是——

最小深度 == Math.min(左子树最小深度,右子树最小深度) + 1

因为最小深度对应一条路径,既然左右两棵子树都有一条最短路径,那么根节点的最短路径肯定就是子树最短路径再加上1。

递推终止条件怎么写呢?我们知道总有一个时候会递归到null节点,null节点没有左右子树了,不能递归了,应该终止递归,并返回最小深度0.

if(root == null) return 0;

这是最初始的、条件反射的思考。一个大致的递推框架建成后,再考虑特殊情况。为什么说有特殊情况?因为我们是依照上面的图思考的,在上面的图中,节点5的左右子树都不为null;为了使递推公式更严谨,我们还需要考虑左右子树为null的情况,如下图,如果左右子树有一个为null,那还能使用Math,min吗?

显然不能,如果对节点1使用Math.min,它的左子树null又返回了结果0,那这就不对了,因为一条最短路径的终点并不是null,也不是节点1,而应该是叶子结点。所以如果出现了有一边子树为Null而另一边不为Null的情况,就要重新写递推式子,这时最小深度应该是:如果左子树为空,最小深度为右子树最小深度+1;如果右子树为null,那么最小深度为左子树最小深度+1。

我们已经考虑了左子树右子树有一边是null的情况,也考虑了左右子树都不为Null的情况,那都为Null呢?都为Null不就是叶子节点吗,叶子结点最小深度为1。只是这个没有必要再单独拉出来写成一个if判断条件,因为用Math.min(左子树最小深度,右子树最小深度)+1来算,叶子结点深度算出来就是1。所以包含在了公式中,不需额外讨论。

综上所述,代码应该这么写,注意递归终止条件写在前面,递推公式写在后面:

public static int minDepth(TreeNode root){
    if (root == null){
        return 0;
    }

    if (root.left == null && root.right != null){
        return minDepth(root.right) + 1;
    }

    if (root.left != null && root.right == null){
        return minDepth(root.left) + 1;
    }

    return Math.min(minDepth(root.left),minDepth(root.right)) + 1;
    //如果根节点不为Null,左右子树也不是只有一边为null,就是正常情况,返回左右子树最小深度中的最小值加1。
    //这里没有单独列出叶子结点的判断,是因为如果是叶子结点就返回1,跟用叶子结点的两个null子树做递归是一样的。
}

还有没有别的方法呢?是有的,你想,最小深度对应路径是根节点和叶子结点路径中,最短的一条。实际上不就是找隔根节点层数最少的叶子结点吗?也就是寻找最上层的叶子结点。那么使用层序遍历的方式一层一层地遍历节点,找到第一个叶子结点就返回当前层数,这个层数就是最小深度,不信可以画图自己看看。

代码如下:

public static int minDepth2(TreeNode root){
        if (root == null){
            return 0;
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int res = 0;
        //res代表结果,是第多少层。
        //这里先使用层序遍历,对于每个出队的元素都判断是否为叶子结点,一旦是就返回当前层数。
        while (!queue.isEmpty()){
            int size = queue.size();
            res++;//res为当前层数,while循环每进行一次都相当于开始遍历下一层,因此每次循环res++
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.remove();
                //判断是否为叶子结点,是就直接返回层数
                if (node.left == null && node.right == null){
                    return res;
                }
                if (node.left != null){
                    queue.add(node.left);
                }
                if (node.right != null){
                    queue.add(node.right);
                }
            }
        }
        return 0;
    }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值