题目:给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例:
注意:两结点之间的路径长度是以它们之间边的数目表示。
思路:
目标:求出二叉树最长直径
要解决的两个问题:
1、如何求出两个结点之间的直径
2、如何找到最长直径
从已给出的二叉树代码中得知,从一个结点只能得到它的左孩子和右孩子,而得不出它的父结点。
若求 4到5的直径长度
不能说从4推到2再到5。只能从父结点2开始,求2到4的直径长度,再求2到5的直径长度,二者之和就是4到5的直径长度。
总结:
求两个结点间的直径,则从父结点开始,左子树的深度 + 右子树的深度 就是两个结点的直径长度。
得出函数返回语句:return depth(左子树) + depth(右子树);
该函数解决的是求父节点的最长直径。
函数代码为:
那么将二叉树的根节点传入函数,则可求出二叉树最长直径。
static int max = 0;
public int diameterOfBinaryTree(TreeNode root) {
max = 0;
depth(root);
return max;
}
static int depth(TreeNode root) {
if (root==null) {
return 0;
}
int L,R;
L = depth(root.left);
R = depth(root.right);
max = (L+R)>max?(L+R):max;
return L>R?L+1:R+1;
}