刷题笔记-543. 二叉树的直径

73 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第21天,点击查看活动详情

一、题目描述:

543. 二叉树的直径 - 力扣(LeetCode)

给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :

给定二叉树

          1
         / \
        2   3
       / \     
      4   5   

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意:两结点之间的路径长度是以它们之间边的数目表示。

二、思路分析:

一种无需全局变量保存当前最大值的思路。

一个根节点下的直径有3种可能:
1、直径在左子树内部,不经过根节点
2、直径在右子树内部,不经过根节点
3、直径经过当前根节点,且直径为两个子树深度之和

分别对左右子树递归调用,返回左右子树的直径和深度
那么当前节点的直径为 max(左直径,右直径,左深度+右深度),当前节点深度为 max(左深度,右深度)+1
再将直径和深度返回给上一层

递归结束的条件是到达空节点,直接返回 0,0

三、AC 代码:

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

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:

        def recursion(node):
            if not node:
                return 0,0
            max_len_left,deep_left = recursion(node.left)
            max_len_right,deep_right = recursion(node.right)
            manx_len = max(max_len_left, max_len_right, deep_left+deep_right)
            deep = max(deep_left,deep_right) + 1
            return manx_len,deep

        return recursion(root)[0]

范文参考:

in rust we trust -- tree (diameter_of_binary_tree/543. 二叉树的直径) - 二叉树的直径 - 力扣(LeetCode)

思路就是遍历每个节点,计算每个节点的左子树最大深度+右子树最大深度设为当前节点的直径长度,要求的就是所有节点直径长度中最大的那个值 - 二叉树的直径 - 力扣(LeetCode)