力扣543. 二叉树的直径

给你一棵二叉树的根节点，返回该树的 直径 。二叉树的直径是指树中任意两个节点之间最长路径的长度 。这条路径可能经过也可能不经过根节点 root 。两节点之间路径的 长度 由它们之间边数表示。

示例 1：

输入：root = 1,2,3,4,5 输出：3

解释：3 ，取路径 4,2,1,3 或 5,2,1,3 的长度。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int ret;
    //计算以node为根的子树的深度和直径
    int dfs(TreeNode* node){
        //因为统计的是路径长度，所以叶子节点深度为0，空节点深度为-1
        if(!node){return -1;}
        int ldepth=dfs(node->left);//左子树最大深度
        int rdepth=dfs(node->right);//右子树最大深度
        //两个叶子分别在左右两棵子树内，直径是多少。尝试更新最大直径
        ret=max(ret,ldepth+rdepth+2);
        return max(ldepth,rdepth)+1;//返回深度
    }
    int diameterOfBinaryTree(TreeNode* root) {
        ret=0;
        dfs(root);
        return ret;
    }
};