Python实现"二叉树的最小深度"的两种方法

最新推荐文章于 2024-03-25 12:57:59 发布

原创最新推荐文章于 2024-03-25 12:57:59 发布 · 1.4k 阅读

5 ·

CC 4.0 BY-SA版权

Algorithms 专栏收录该内容

106 篇文章

订阅专栏

本文介绍两种求解二叉树最小深度的有效算法。一种通过逐层遍历找到最近的叶子节点；另一种采用递归方式处理左右子树，特别考虑只有一个子树的情况。这两种方法都适用于LeetCode上的相关题目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

找到给定二叉树的最小深度

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

注意：叶子节点没有子树

Example:

Given binary tree [3,9,20,null,null,15,7],
    3
   / \ 
  9  20
    /  \
   15   7
return its minimum depth = 2.

1：算法遍历二叉树每一层，一旦发现某层的某个结点无子树，就返回该层的深度，这个深度就是该二叉树的最小深度

def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        curLevelNodeList = [root]
        minLen = 1
        while curLevelNodeList is not []:
            tempNodeList = []
            for node in curLevelNodeList:
                if not node.left and not node.right:
                    return minLen
                if node.left is not None:
                    tempNodeList.append(node.left)
                if node.right is not None:
                    tempNodeList.append(node.right)
            curLevelNodeList = tempNodeList
            minLen += 1
        return minLen

2：用递归解决该题和"二叉树的最大深度"略有不同。主要区别在于对“结点只存在一棵子树”这种情况的处理，在这种情况下最小深度存在的路径肯定包括该棵子树上的结点

def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        if not root.left and root.right is not None:
            return self.minDepth(root.right)+1
        if root.left is not None and not root.right:
            return self.minDepth(root.left)+1
        left = self.minDepth(root.left)+1
        right = self.minDepth(root.right)+1
        return min(left,right)

算法题来自：https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/description/