题目:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
其实这道题和这道题有点类似
且听我慢慢道来:
算法分析:
很显然,这道题用递归求解。无非就是求每个节点的左节点深度和右节点深度和,然后比较大小求最大值!
以图示为例:
节点1:左子树高度为2,右子树高度1,所以和为2+1+1=4
节点2:左子树高度为1,右子树高度1,所以和为1+1+1=3
节点3、4、5:左子树右子树高度均为0,所以和为0+0+1=1
所以最大值为4,一共经过四个节点,也即其之间数目为4-1=3
那么写个递归函数depth(root)是用来求解每个节点的深度,那么左子树高度为depth(root.left),右子树高度为depth(root.right),所以其和为depth(root.left)+depth(root.right)+1
取一个变量max保存遍历期间的最大值。
最后结果需要为max-1(因为求的是边的数目!)
代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
int max = 0;
public int diameterOfBinaryTree(TreeNode root) {
depth(root);
return max-1;
}
public int depth(TreeNode root){
if(root==null){
return 0;
}
int left = depth(root.left);
int right =depth(root.right);
max = (left+right)+1>max?(left+right)+1:max;
return Math.max(left,right)+1;
}
}