Python求二叉树的属性（深度、直径等）

最新推荐文章于 2024-04-07 13:21:37 发布

化身孤岛的鲸o

最新推荐文章于 2024-04-07 13:21:37 发布

阅读量306

点赞数

CC 4.0 BY-SA版权

分类专栏：树文章标签： python leetcode

本文链接：https://blog.csdn.net/weixin_44123362/article/details/129773555

树专栏收录该内容

39 篇文章

订阅专栏

文章介绍了如何使用递归方法在LeetCode的两道题目中求解二叉树的最大深度和直径。对于最大深度，通过计算左右子树的最大深度并取最大值加1来得到。而对于直径，需要两次深度优先搜索(DFS)分别计算左右子树的深度，并找到最长路径的长度。

递归得到二叉树的最大深度

参考题目

LeetCode 104. 二叉树的最大深度

实现代码

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

class Solution:
    def maxDepth(self, root):
        if root is None: 
            return 0 
        else: 
            left_height = self.maxDepth(root.left) 
            right_height = self.maxDepth(root.right) 
            return max(left_height, right_height) + 1

递归得到二叉树的直径

参考题目

LeetCode 543. 二叉树的直径

实现代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        def getTreeDepth(root):
            if not root:
                return 0
            leftDepth = getTreeDepth(root.left)
            rightDepth = getTreeDepth(root.right)
            return max(leftDepth, rightDepth) + 1
        
        def dfs(root):
            if not root:
                return
            leftDepth, rightDepth = 0, 0
            if root.left:
                leftDepth = max(leftDepth, getTreeDepth(root.left))
            if root.right:
                rightDepth = max(rightDepth, getTreeDepth(root.right))
            nonlocal res
            res = max(res, leftDepth + rightDepth)
            if root.left:
                dfs(root.left)
            if root.right:
                dfs(root.right)
        
        res = 0
        dfs(root)
        return res