leetcode-543. 二叉树的直径

manba_

已于 2025-02-02 21:35:54 修改

阅读量325

点赞数 5

分类专栏： leetcode hot100 文章标签： leetcode 算法

于 2024-08-20 20:28:06 首次发布

本文链接：https://blog.youkuaiyun.com/m0_38098373/article/details/141367560

版权

leetcode hot100 专栏收录该内容

100 篇文章

订阅专栏

题目描述

给你一棵二叉树的根节点，返回该树的直径。

二叉树的直径是指树中任意两个节点之间最长路径的长度。这条路径可能经过也可能不经过根节点 root 。

两节点之间路径的长度由它们之间边数表示。

示例 1：

输入：root = [1,2,3,4,5]
输出：3
解释：3 ，取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

示例 2：

输入：root = [1,2]
输出：1

思路

如果做过leetcode-104. 二叉树的最大深度-优快云博客这道题，很容易可以做出本题

无非在求最大深度的代码中，加入了一个求直径的过程。直径等于左子树深度+右子树深度+1，每次和预先设置好的最大值进行比对

PS：最后返回的是self.res-1，因为我们记录的一直都是节点的个数，直径等于节点个数-1，求边的个数

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def __init__(self):
        self.res = 0
    def depth(self, root):
        if not root:
            return 0 
        ldepth = self.depth(root.left)
        rdepth = self.depth(root.right)
        self.res = max(ldepth+rdepth+1, self.res)
        return max(ldepth,rdepth)+1
    def diameterOfBinaryTree(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        self.depth(root)
        return self.res-1

二刷：

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

    def heigh(self,root):
        if not root:
            return 0 
        ldepth = self.heigh(root.left)
        rdepth = self.heigh(root.right)
        self.max_len = max(self.max_len, ldepth+rdepth)
        return max(ldepth,rdepth)+1

    def diameterOfBinaryTree(self, root):
        """
        :type root: Optional[TreeNode]
        :rtype: int
        """
        if not root:
            return 0
        self.heigh(root)
        return self.max_len

记忆点：

树的深度那题修改：直径=左深度+右深度+1