给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
这里最长直径的边一定都是叶子结点,用深度遍历整个树。一般思路是找出节点左边的最大长度和右边的最大长度加起来可能就是最大直径,这里就只需要返回 max(right,left)。但是因为最长直径可能不过根节点,所以这里多加一个判断条件,如果当前节点的左子树最大长度和右子树最大长度之和大于当前最大长度就更新一下。
class Solution {
int max = 0;
public int diameterOfBinaryTree(TreeNode root) {
if(root==null)return 0;
dfs(root);
return max-2; //叶子结点处多加了1,两个顶点就多加了2
}
public int dfs(TreeNode node){
if(node==null)return 0;
int left = bfs(node.left)+1;
int right = bfs(node.right)+1;
if(left+right>max)max=left+right;
return Math.max(left,right);
}
}