543. 二叉树的直径
题目:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
注意:这条路径可能穿过也可能不穿过根结点。
class Solution {
public:
int diameterOfBinaryTree(TreeNode* root) {
if(root == nullptr)
return 0;
int d1 = maxDepth(root->left) + maxDepth(root->right);
int d2 = diameterOfBinaryTree(root->left);
int d3 = diameterOfBinaryTree(root->right);
int d = d1;
if(d2 > d)
{
d = d2;
}
if(d3 > d)
{
d = d3;
}
return d;
}
int maxDepth(TreeNode* root)
{
if(root == nullptr)
return 0;
int maxLeftDepth = maxDepth(root->left);
int maxRightDepth = maxDepth(root->right);
return maxLeftDepth > maxRightDepth ? maxLeftDepth + 1 : maxRightDepth + 1;
}
};
一开始只考虑了d1这一种情况,
也有d2或d3这种情况,如下图: