力扣-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++中使用引用&的写法。