543. 二叉树的直径

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

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。

解题思路:
二叉树的直径可以转化成二叉树的左右子树高度之和。
求直径长度的最大值即为求左右子树高度之和的最大值。
maxPath = max(init, LHeight + RHeight)。

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var diameterOfBinaryTree = function(root) {
    if(!root) return 0;
    let max = 0; 

    function deep(root){
        if(!root) return 0;
        let leftDeep = deep(root.left);
        let rightDeep = deep(root.right);
        max = Math.max(max,leftDeep + rightDeep);
        return Math.max(leftDeep,rightDeep) + 1;
    };

    deep(root);
    return max;
};

复杂度分析

  • 时间复杂度:O(N),其中 N 为二叉树的节点数,即遍历一棵二叉树的时间复杂度,每个结点只被访问一次。
  • 空间复杂度:O(Height)。其中Height 为二叉树的高度。由于递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,而递归的深度显然为二叉树的高度,并且每次递归调用的函数里又只用了常数个变量,所以所需空间复杂度为O(Height) 。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树直径是指任意两个节点之间最长路径的长度。在 Python 中,可以通过递归的方式来计算二叉树直径。以下是一个示例代码: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def diameterOfBinaryTree(root): if not root: return 0 # 递归计算左子树和右子树的最大深度 left_depth = maxDepth(root.left) right_depth = maxDepth(root.right) # 计算当前节点的直径 current_diameter = left_depth + right_depth # 递归计算左子树和右子树的直径 left_diameter = diameterOfBinaryTree(root.left) right_diameter = diameterOfBinaryTree(root.right) # 取最大直径 return max(current_diameter, left_diameter, right_diameter) def maxDepth(root): if not root: return 0 else: left_depth = maxDepth(root.left) right_depth = maxDepth(root.right) return max(left_depth, right_depth) + 1 # 构建二叉树 root = TreeNode(1) root.left = TreeNode(2) root.right = TreeNode(3) root.left.left = TreeNode(4) root.left.right = TreeNode(5) diameter = diameterOfBinaryTree(root) print(f"The diameter of the binary tree is {diameter}") ``` 上述代码中,首先定义了一个 `TreeNode` 类来表示二叉树的节点。然后,定义了 `diameterOfBinaryTree` 函数来计算二叉树直径。 在 `diameterOfBinaryTree` 函数中,通过递归计算左子树和右子树的最大深度,并计算当前节点的直径。然后,再递归计算左子树和右子树的直径,并取最大值作为最终的直径。 另外,还定义了 `maxDepth` 函数来计算二叉树的最大深度。 运行以上代码,将会输出以下结果: ``` The diameter of the binary tree is 3 ``` 表示给定的二叉树直径为 3。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值