题目描述:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
**注意:**两结点之间的路径长度是以它们之间边的数目表示。
方法1:
主要思路:
(1)最大路径可能存在中间某个结点的位置,故需要一直保留每个节点处可以获得的最大的直径;
(2)对于辅助函数的返回值,则是在该节点下可以获得的最大的深度是多少;
(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 helper(TreeNode* root,int& max_path){
if(root==NULL){
return 0;
}
//获得左右子树的最大深度
int left_path=helper(root->left,max_path);
int right_path=helper(root->right,max_path);
max_path=max(max_path,left_path+right_path);//获得当前结点下可能获得的最大的直径
return max(left_path,right_path)+1;//返回该结点下可以获得最大的深度
}
int diameterOfBinaryTree(TreeNode* root) {
if(root==NULL){
return 0;
}
int max_path=0;
helper(root,max_path);
return max_path;
}
};