1 题目
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diameter-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2 Java
2.1 方法一(错误解法)
误认为最长直径就是以根节点为中间,左侧深度 + 右侧深度
极端情况
根节点无左子树,右子树是一个左右深度各为 5 的树
那么以根节点为中间节点,直径为 5
以右孩子为中间节点,直径为 8
显然解法不对
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int diameterOfBinaryTree(TreeNode root) {
if(root == null) return 0;
return deepth(root.left) + deepth(root.right);
}
// 计算某节点的深度
public int deepth(TreeNode root){
if(root == null) return 0;
int leftH = deepth(root.left);
int rightH = deepth(root.right);
int rootH = Math.max(leftH, rightH) + 1;
return rootH;
}
}
2.2 方法二(正确解法,递归)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int diameterOfBinaryTree(TreeNode root) {
deepth(root);
return ans;
}
// 用于记录最长直径的值
int ans = 0;
// 计算某节点的深度
public int deepth(TreeNode root){
if(root == null) return 0;
int lH = deepth(root.left);
int rH = deepth(root.right);
int H = Math.max(lH, rH) + 1;
// 仅仅多了这一步!!!计算以root节点为中间节点的直径,并更新ans
ans = Math.max(ans, lH + rH);
// 每次递归返回的任然是深度,而非直径
return H;
}
}