给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。
示例 1:
输入:root = [1,2,3,4,5]
输出:3
解释:3 ,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。
示例 2:
输入:root = [1,2]
输出:1
提示:
树中节点数目在范围 [1, 104] 内
-100 <= Node.val <= 100
/**
* 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;
* }
* }
*/
class Solution {
int max=0;
public int diameterOfBinaryTree(TreeNode root) {
if (root == null) return 0;
deep(root);
return max;
}
private int deep(TreeNode root) {
if (root == null) return 0;
// 递归计算左子树和右子树的最大深度
int leftDepth = deep(root.left);
int rightDepth = deep(root.right);
max=Math.max(max,leftDepth+rightDepth);
// 当前节点的最大深度是左子树和右子树深度的最大值加一
return 1 + Math.max(leftDepth, rightDepth);
}
}
我陷入了一个死胡同,认为只要求出左子树的最大深度+右子树的最大深度,就可以得出直径,这就存在一个误区,就是直径的路径不一定存在在根节点的一级子树中,如下图所示,直径路径存在在-9为根节点的子树中,所以必须递归对子树的直径进行更新,即要寻找的是这棵树中左右子树相加最大的值,而不是直接求左右子树的深度。