543. 二叉树的直径

 

 使用DFS求解

时间O(n)，空间O(h)(h为整棵树的深度)

class Solution {
    int res=0;
    public int diameterOfBinaryTree(TreeNode root) {
        def(root);
        return res;
    }

    public int def(TreeNode root){
        if(root==null) return 0;
        int left=def(root.left);
        int right=def(root.right);
        // 计算左右子树高度和，本题要求路径和，刚好不需要额外处理
        res = Math.max(left+right,res);
        // 计算本子树的高度
        return Math.max(left,right)+1;
    }
}