【leetcode】543 二叉树的直径（二叉树，递归）

题目链接：https://leetcode-cn.com/problems/diameter-of-binary-tree/submissions/

题目描述

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。

示例 :

给定二叉树

          1
         / \
        2   3
       / \     
      4   5    

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意：两结点之间的路径长度是以它们之间边的数目表示。

代码

class Solution {
public:
    int diameterOfBinaryTree(TreeNode* root) {
        if(root == nullptr) return 0;
        int ret = INT_MIN;
        maxDiameter(root, ret);
        return ret;
    }

    int maxDiameter(TreeNode* root, int &ret){
        if (root == nullptr) return 0;
        int leftVal = maxDiameter(root->left, ret);
        int rightVal = maxDiameter(root->right, ret);

        int local = max(leftVal, rightVal) + 1;
        ret = max(ret, leftVal + rightVal);
        return local;
    }
};