543. 二叉树的直径

543. 二叉树的直径

描述

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

示例 :
给定二叉树

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

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

思路

首先我们知道一条路径的长度为该路径经过的节点数减一,所以求直径(即求路径长度的最大值)等效于求路径经过节点数的最大值减一。

目的:自上而下,深度优先遍历,得到每一个节点的长度,左右子节点两边的长度之和加一即为其父节点的“长度”,以此遍历得到每个节点的“长度”,维护一个变量,实时更新最长的“长度”,并返回该节点为根的子树的深度。其中,basecase:if (node == null) return 0;
(任意一条路径均可以被看作由某个节点为起点,从其左儿子和右儿子向下遍历的路径拼接得到 -->递归)

在这里插入图片描述

class Solution {
    int ans;
    public int diameterOfBinaryTree(TreeNode root) {
        ans = 1;
        depth(root);
        return ans - 1;
    }
    public int depth(TreeNode node) {
        if (node == null) return 0; // 访问到空节点了,返回0--->basecase
        int L = depth(node.left); // 左儿子为根的子树的深度
        int R = depth(node.right); // 右儿子为根的子树的深度
        ans = Math.max(ans, L+R+1); // 计算d_node即L+R+1 并更新ans
        return Math.max(L, R) + 1; // 返回该节点为根的子树的深度
    }
}

复杂度分析

  • 时间复杂度:O(N),其中 N 为二叉树的节点数,即遍历一棵二叉树的时间复杂度,每个结点只被访问一次。

  • 空间复杂度:O(Height),其中Height 为二叉树的高度。由于递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,而递归的深度显然为二叉树的高度,并且每次递归调用的函数里又只用了常数个变量,所以所需空间复杂度为 O(Height) 。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
二叉树直径是指任意两个节点之间最长路径的长度。在 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。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值