543. 二叉树的直径

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

二叉树的直径不一定过根节点,因此需要去搜一遍所有子树(例如以root,root.left, root.right…为根节点的树)对应的直径,取最大值。
root的直径 = root左子树高度 + root右子树高度
root的高度 = max {root左子树高度, root右子树高度} + 1

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
 //使用dfs
class Solution {
    int max=0;
    public int dfs(TreeNode root){
      if(root==null){
          return 0;
      }  
      int left = dfs(root.left);
      int right = dfs(root.right);
      max = Math.max(left+right,max);
      return Math.max(left,right)+1;

    }
    public int diameterOfBinaryTree(TreeNode root) {
        dfs(root);
        return max;
    
    }
}
  • 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
发出的红包

打赏作者

<编程路上>

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值