思路
- 要得到左右子树的最大深度。
- h为当前的最大深度,max_h为当前最大宽度
代码
#define max(a,b) ((a)>(b)?a:b)
void treehigh(struct TreeNode *root,int *h,int *max_h){
if(root==NULL){
*h=0;*max_h=0;
return;
}
int h1,h2;
int max_1,max_2;
treehigh(root->left,&h1,&max_1);
treehigh(root->right,&h2,&max_2);
*h = max(h1, h2)+1;//当前节点的最大深度
*max_h = max(h1+h2, max(max_1, max_2));//当前节点的最大宽度,要么是左右子树的高之和,要么是左右子树的最大宽度
}
int diameterOfBinaryTree(struct TreeNode* root){
int h = 0,max_h=0;
treehigh(root,&h,&max_h);
return max_h;
}