求二叉树的最小深度(普通+Morris遍历进阶)

本文介绍了如何使用二叉树的普通解法和Morris遍历求解最小深度,详细阐述了Morris遍历的原理,并解释了如何在Morris遍历过程中发现所有叶节点,实现时间复杂度O(n),空间复杂度O(1)。
摘要由CSDN通过智能技术生成

求一棵二叉树的最小深度


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

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。

leetcode题目跳转

普通解法二叉树的递归套路

时间复杂度O(n),需要遍历所有的节点。
空间复杂度为O(h)。其中,h 表示二叉树的高度,也就是递归使用的系统调用栈的高度。

  记头节点为 root

  1. 左子树为空,求右子树的最小高度 + 1
  2. 右子树为空,求左子树的最小高度 + 1
  3. 左、右子树不为空,求左子树的最小高度 + 1,求右子树的最小高度 + 1
  4. 左、右子树都为空,最小高度为 1
class Solution {
   
    public int minDepth(TreeNode root) {
   
        if (root != null) {
   
            if (root.left == null || root.right == null) {
   // 只有一个孩子或没有孩子
                return Math.max(minDepth(root.left), minDepth(root.right)) + 1;
            } else {
    // 有两个孩子
                return Math.min(minDepth(root.left), minDepth(root.right)) + 1;
            }
        } else return 0;
    }
}

进阶:Morris遍历改写

时间复杂度O(n),需要遍历所有的节点。
空间复杂度为O(1)

Morris遍历详细讲解

思路:

  首先,解决在morris遍历中得到每一个当前节点的深度。

  遍历的当前节点记为 cur,cur的深度是 level

  morris遍历中,下一个节点深度是多少?

morris遍历的规则:

  1. 如果cur没有左子树,cur向右移动(cur = cur.right),那么下一个节点就是cur的右孩子,深度:level+1。

  2. 如果cur有左子树,找到cur左子树上最右的节点,记mostRight。

    ①如果 mostRight 的 right 指针指向null,让其指向cur(mostRight.right = cur),cur向左移动(cur =.cur.left),那么下一个节点就是cur的左孩子,深度:level+1。

    ②如果 mostRight 的right 指针指向 cur,让其指向空( mostRight.right = null),cur向右移动(cur= cur.right)假设下一个节点记为next,根据 morris遍历可知,cur是next左子树上最右的节点。next 的深度 = level - next左子树的右边界的节点数。

利用以上策略,就能在morris遍历中得到每一个节点的深度。

其次,解决如何在morris遍历中发现每一个叶节点

  因为在morris遍历中,我们会人为修改某些节点的 right 指针,让其指向上级的某个节点。这样,当用morris遍历到某个节点×的时候,也许×原本是叶节点,但此时却发现不了,因为此时X的 right 指针指向上级了(不满足X.left == null && X.right = null)。
所以,为了发现所有的叶节点,我们把发现叶节点的时机放在morris遍历中回到自己两次,且第二次回到这个节点的时候。
在这里插入图片描述

比如,使用Mirror遍历二叉树时(上图所示):
  当第二次回到2的时候,看看1是不是叶节点;
  当第二次回到4的时候,看看3是不是叶节点;
  当第二次回到6的时候,看看5是不是叶节点;
  最后单独看一下整棵树的最右节点是不是叶节点。

  这样就能在morris遍历中找到所有的叶节点了。做到了在morris遍历中,每一个节点的深度都能得到;也做到了在morris遍历中发现所有的叶节点。这个问题自然可以求解,遍历的代价就是morris 遍历的代价,时间复杂度为O(N),额外空间复杂度O(1)。

	// 根据 morris 遍历改写
	public static int minHeight
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值