直径即左右子树最大深度之和
递归的方法求出左子树的最大深度和又子树的最大深度
通过定义一个全局变量 sum ,与a+b(当前左右子树最大的深度)比较,不断更新sum的值,最后即为直径的长度
/**
* 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) {
height(root);
return sum;
}
private :
int sum = 0;
int height(TreeNode* root)
{
if(!root)
return 0;
int a = height(root->left);//找出左子树深度最大的地方,如果为叶子节点,则返回1 (算上叶子节点的深度)
int b = height(root->right);
sum = max(sum, a + b);//a+b与sum总计的值对比,选取其中大的,如果a+b比sum大则更新sum
return max(a,b)+1; //算上本节点的深度
}
};