1,题目要求
Given a binary tree, you need to compute the length of the diameter of the tree. The diameter of a binary tree is the length of the longest path between any two nodes in a tree. This path may or may not pass through the root.
Example:
Given a binary tree as follow:
Return 3, which is the length of the path [4,2,1,3] or [5,2,1,3].
Note: The length of path between two nodes is represented by the number of edges between them.
也就是找到一颗二叉树中的最长路径的大小。
2,题目思路:
需要注意的是,找到的最长路径未必就是整棵树的左子树高度加上右子树的高度,因为有可能左子树的节点数目众多,使得左子树中存在最长路径,而右子树可能就一个节点。
因此,出了需要利用一个递归的函数来求书的高度外,还要对左右子树中的最长路径进行计算,最后去max。
C++中有内置的max函数,但是最多只能判断两个值之间的最大值(没有python方便),因此需要利用两个max函数来求其中的最大的值。当然,当值的数量过多时,可以利用数组进行存储并利用sort函数进行排序,得到最大的值。
3,程序源码:
/**
* 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:
int diameterOfBinaryTree(TreeNode* root) {
int res = 0;
if(root == nullptr)
return 0;
else
res = MaxDeepth(root->left) + MaxDeepth(root->right);
return max(res,max(diameterOfBinaryTree(root->left),diameterOfBinaryTree(root->right)));
}
public:
int MaxDeepth(TreeNode* root){
if(root == nullptr)
{
return 0;
}
return 1 + max(MaxDeepth(root->left),MaxDeepth(root->right));
}
};