每日一练:二叉树的直径
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;
};