class Solution {
int ans = 0;
public int diameterOfBinaryTree(TreeNode root) {
dfs(root);
return ans;
}
public int dfs(TreeNode root) {
if(root==null)return 0;
int r = dfs(root.right);
int l = dfs(root.left) ;
ans = Math.max(ans,r+l); //根节点点加左右
return Math.max(1+r,1+l); //看子节点那边比较大 就返回那个
}
}
最大的直径可能有三种情况:
1.在左子树内部
2.在右子树内部
3.在穿过左子树,根节点,右子树的一条路径中
设计一个递归函数,返回以该节点为根节点,向下走的最长路径
知道这个值以后
以某个节点为根节点的最长直径就是,该节点左子树向下走的最长路径 ,再加上该节点右子树向下走的最长路径
我们用一个全局变量记录这个最大值,不断更新