543. 二叉树的直径 - 力扣(LeetCode)
一、题目要求
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root
。
两节点之间路径的 长度 由它们之间边数表示。
示例 1:
输入:root = [1,2,3,4,5] 输出:3 解释:3 ,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。
示例 2:
输入:root = [1,2] 输出:1
提示:
- 树中节点数目在范围
[1, 104]
内 -100 <= Node.val <= 100
二、解法1-递归 O(N)
因为题目说到最长路径可能不经过根节点,所以递归每个节点的左右子树,得到它们的高度,如果加起来大于ret,就更新ret,然后返回上一个节点的高度。
class Solution {
int height(TreeNode* root)
{
if(root == nullptr)
return 0;
int leftHeight = height(root->left);
int rightHeight = height(root->right);
if(leftHeight+rightHeight > ret)
ret = leftHeight+rightHeight;
return max(leftHeight,rightHeight)+1;
}
public:
int diameterOfBinaryTree(TreeNode* root) {
height(root);
return ret;
}
private:
int ret = 0;
};