一、题目
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
注意:两结点之间的路径长度是以它们之间边的数目表示。
二、思路
与二叉树的深度相结合。
两个节点之间的路径长度是它们之间边的数目,所以,一条路径的长度就等于该路径经过的节点数减一。而一条路径可以看成是以某个节点为根,再加上左右子树组成的。
我们又知道树的深度是以节点数表示的,若以某个节点为根,左子树深度为 L,右子树深度为 R,则以该节点和其左右子树组成的路径的节点数为(L + R + 1),从而该路径的长度为(L + R + 1)- 1。
所以,遍历整棵树,定义一个变量 t,记录以每个节点组成的路径的节点数的最大值,最后返回 t - 1 即为树的直径。
三、代码
class Solution {
int t;
public int diameterOfBinaryTree(TreeNode root) {
t = 1;
depth(root);
return t - 1;
}
// 求二叉树的最大深度
public int depth(TreeNode root){
if(root == null)
return 0;
int l = depth(root.left);// 左子树的深度
int r = depth(root.right);// 右子树的深度
t = (t > l + r + 1 ? t : l + r + 1);// 求出当前路径的节点数,即(l + r + 1),并更新 t
return l > r ? (l + 1) : (r + 1);
}
}