题目:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
思路:
任意一条路径均可以被看作由某个节点为起点,从其左儿子和右儿子向下遍历的路径拼接得到。
假设我们知道对于该节点的左儿子向下遍历经过最多的节点数 L(即以左儿子为根的子树的深度) 和其右儿子向下遍历经过最多的节点数 R (即以右儿子为根的子树的深度),那么以该节点为起点的路径经过节点数的最大值即为 L+R+1 。
流程:定义递归函数,返回该节点为根的子树的深度。先递归调用左儿子和右儿子求得它们为根的子树的深度 L 和 R ,
1.该节点为根的子树的深度即为 max(L,R)+1。
2。该节点的直径值为 L+R+1。
递归搜索每个节点并设一个全局变量 ans 记录 直径 的最大值,最后返回
ans-1
即为树的直径。
class Solution {
int ans;
public int diameterOfBinaryTree(TreeNode root) {
ans=1;
dfs(root);
return ans-1;
}
public int dfs(TreeNode node){
if(node==null) return 0;
int l=dfs(node.left);
int r=dfs(node.right);
ans=Math.max(ans,l+r+1);
return Math.max(l,r)+1;
}
}