给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。
示例 :
给定二叉树
1 / \ 2 3 / \ 4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
题目标签:Tree
题目链接:LeetCode / LeetCode中国
Language | Runtime | Memory |
---|---|---|
cpp | 12 ms | N/A |
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> tmp;
int maxDepth(TreeNode* node){
if(!node){
return 0;
}
int l = maxDepth(node->left);
int r = maxDepth(node->right);
tmp.push_back(l + r);
return 1 + max(l, r);
}
int diameterOfBinaryTree(TreeNode* root) {
if(!root){
return 0;
}
maxDepth(root);
return *max_element(tmp.begin(), tmp.end());
}
};