持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第21天,点击查看活动详情
一、题目描述:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树
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)