力扣-543.二叉树的直径

题目描述

给你一棵二叉树的根节点，返回该树的 直径 。

二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。

两节点之间路径的 长度 由它们之间边数表示。

class Solution {
public:
    int maxLength(TreeNode *root, int &ans) {
        if (!root) {
            return 0;
        }
        int left = maxLength(root->left, ans);
        int right = maxLength(root->right, ans); //这里不用left和right提前算出来会超时
        ans = max(ans, left + right);
        return max(left, right) + 1;
    }
    int diameterOfBinaryTree(TreeNode *root) {
        int ans = 0;
        maxLength(root, ans);
        return ans;
    }
};

小结：这道题感觉依然有中等题的难度，需要转化一下思维，直径相当于左子树最大高度+右子树最大高度，同时注意这种需要在递归中不断更新并回收一个数，在C++中使用引用&的写法。