LeetCode543题:二叉树的直径(python3)

代码思路:

先递归调用左儿子和右儿子求得它们为根的子树的深度 L和 R ,则该节点为根的子树的深度即为max(L,R)+1。该节点的 dnode值为L+R+1

递归搜索每个节点并设一个全局变量 ans记录 dnode的最大值,最后返回 ans-1 即为树的直径。

python 复制代码
# 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:
        self.ans = 1
        def depth(node):
            # 访问到空节点了,返回0
            if not node: 
                return 0 
            # 左儿子为根的子树的深度
            L = depth(node.left) 
            # 右儿子为根的子树的深度
            R = depth(node.right)
            # 计算d_node即L+R+1 并更新ans
            self.ans = max(self.ans,L+R+1)
            # 返回该节点为根的子树的深度
            return max(L,R)+1 
        depth(root)
        return self.ans-1
相关推荐
NAGNIP8 小时前
轻松搞懂全连接神经网络结构!
人工智能·算法·面试
NAGNIP8 小时前
一文搞懂激活函数!
算法·面试
董董灿是个攻城狮8 小时前
AI 视觉连载7:传统 CV 之高斯滤波实战
算法
用户83562907805112 小时前
无需 Office:Python 批量转换 PPT 为图片
后端·python
爱理财的程序媛14 小时前
openclaw 盯盘实践
算法
markfeng814 小时前
Python+Django+H5+MySQL项目搭建
python·django
GinoWi15 小时前
Chapter 2 - Python中的变量和简单的数据类型
python
JordanHaidee15 小时前
Python 中 `if x:` 到底在判断什么?
后端·python
ServBay15 小时前
10分钟彻底终结冗长代码,Python f-string 让你重获编程自由
后端·python