543. 二叉树的直径
原题链接
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
要点
要知道一条路径是以当前节点 左边的长度+右边的长度+1
如 4 <- 2 -> 5 = 3,那么 以节点 1 为连接点,就是 4\5 <- 2 ->3
对于我们来说,并不关心左右两边都是些什么内容,只需要知道左右两边给到中的这个连接点的长度(即有多少个节点)是多少。
解法
我们求出节点个数就能够知道它们的路径长度了
代码片段
//543. 二叉树的直径
public int diameterOfBinaryTree(TreeNode root) {
if (root == null) {//0.终止条件
return 0;
}
int[] result = new int[1];
result[0] = 1; //1.创建存放结果的数组,至少有一个元素。
diameterOfBinaryTreeHelper(root, result);//3.求解最长连接的元素个数。
return result[0] - 1;//2.转化成路径长度。
}
public int diameterOfBinaryTreeHelper(TreeNode root, int[] result) {
if (root == null) {//4.终止条件。
return 0;
}
int lDepth = diameterOfBinaryTreeHelper(root.left, result);//5.左边的长度。
int rDepth = diameterOfBinaryTreeHelper(root.right, result);//6.右边的长度。
result[0] = Math.max(lDepth + rDepth + 1, result[0]);//8.更新结果,看看是否以当前为节点的连接路径长度是否更长了。
return Math.max(lDepth, rDepth) + 1;//7.返回最长的长度到上层,整个过程中一直保证左右两边都是保留最长一条路径。
}
补充说明
- 关于为什么一直在更新 result 結果
要知道我们是后续遍历,我们的返回顺序(绿色箭头)
① -> ② -> ③
那么以 2为根节点的长度 6 < 5 < 4 <- 2 -> 7 >8 >9 >10 >11 = 9
那么以 1为根节点的长度 11 < 10 < 9 <8 < 7 < 2 <- 1 -> 3 = 8
也就是说在返回的时候我们即使知道了其中一边的最大长度,也不能等价于目前这段 最 长 距 离 + N = M a x 最长距离+N = Max 最长距离+N=Max
如 10 + 8 = 18 10 + 8 = 18 10+8=18、但是如果变成 10 + 3 10 + 3 10+3 呢? 显然此时不是最大值,所以不断的保留目前已知的最大结果
结尾
1.博客地址
2.源代码仓库
如果你在代码里看到了用 数字标记的注释 如 //1.xxx 这是我写代码的顺序,希望能给你一点启发。