[LeetCode] 111. Minimum Depth of Binary Tree @ python

最新推荐文章于 2021-03-26 11:04:55 发布

原创最新推荐文章于 2021-03-26 11:04:55 发布 · 177 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode

收录于

leetcode 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一种计算二叉树最小深度的递归算法。通过判断根节点及其左右子节点的状态，递归计算左右子树的最小深度，并返回较小值加一。此算法适用于所有二叉树结构。

一．题目：
给一颗二叉树，求它的最小深度
二．解题思路：
给定根节点，我们只需要计算它的左子树的最小深度和右子树的最小深度的较小值+1．所以我们的递归函数就出来了，代码如下：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        #递归法
        if root == None:
            return 0
        if root.left == None and root.right == None:
            return 1
        if root.left == None and root.right != None:
            return self.minDepth( root.right ) + 1
        if root.left != None and root.right == None:
            return self.minDepth( root.left ) + 1
        return min( self.minDepth( root.left ), self.minDepth( root.right ) ) + 1