题目
- 二叉树的直径
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3]
或者[5,2,1,3]
。
注意:两结点之间的路径长度是以它们之间边的数目表示。
分析
- 借助求二叉树的深度的方式。对于每一个节点,求其左右子树的最大深度,然后对于每一个子节点的左右深度加和,与一个统计树的直径的全局变量
max
作比较,如果该节点的左右子树深度和大于他们,则更新结果。
代码
public class Solution543 {
int max = 0;
public int diameterOfBinaryTree(TreeNode root) {
getDepth(root);
return max;
}
public int getDepth(TreeNode root) {
if (root == null)
return 0;
int l = getDepth(root.left);
int r = getDepth(root.right);
max = Math.max(max, l + r);
return Math.max(l + 1, r + 1);
}
}