543. 二叉树的直径

最新推荐文章于 2024-08-26 20:53:14 发布

幸运小猴子

最新推荐文章于 2024-08-26 20:53:14 发布

阅读量139

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/sandrew0916/article/details/90762574

leetcode 专栏收录该内容

88 篇文章

订阅专栏

本文介绍了一种计算二叉树直径长度的方法，通过深度优先搜索算法，找出任意两个节点间最长路径，该路径可能穿过根节点。代码示例使用Python和C++实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

def diameterOfBinaryTree(self, root):
    diameter = [0]
    self.depth(root, diameter)
    return diameter[0]

def depth(self, root, diameter):
    if not root:
        return 0
    left_height = self.depth(root.left, diameter)
    right_height = self.depth(root.right, diameter)
    diameter[0] = max(diameter[0], left_height + right_height)
    return 1 + max(left_height, right_height)

class Solution {
public:
    int maxdiadepth = 0;
    
    int dfs(TreeNode* root){        
        if(root == NULL) return 0;
        
        int leftdepth = dfs(root->left);
        int rightdepth = dfs(root->right);
        
        if(leftdepth + rightdepth > maxdiadepth) maxdiadepth = leftdepth + rightdepth;
        return max(leftdepth +1, rightdepth + 1);     
    }
    
    int diameterOfBinaryTree(TreeNode* root) {        
        dfs(root);
        
        return maxdiadepth;
    }
};