1、题目描述
给定一棵二叉树,计算这课二叉树的直径长度,即为二叉树任意两个节点间的最长路径。比如:
这课二叉树的最长路径为3。
2、解题思路
使用递归进行求解,每次递归的过程中,先求出以某个节点为树根的二叉树的左子树的最长深度left_len、右子树的最长深度right_len,并在递归函数中用一个变量diameter来保存任意两个节点间的最长路径。在求出左子树的最长深度left_len和右子树的最长深度right_len之后,就可以求出以该节点为根的二叉树的最长路径diameter。具体代码如下:
int diameterOfBinaryTree(TreeNode* root) {
if(root==nullptr)
return 0;
int diameter = 0; //保存直径,即任意两节点间的最长的路径长度
int left_len = 0; //左子树的最长深度
int right_len = 0; //右子树的最长深度
find(root, left_len, right_len, diameter);
return diameter;
}
void find(TreeNode *root,int &left_len,int &right_len,int &diameter){
int left = 0;
int right = 0;
if(root->left){
find(root->left,left,right,diameter);
left_len = left>right? left+1:right+1; //左子树最大深度取最大值 + 1
}
else{
left_len = 0;
}
if(root->right){
find(root->right,left,right,diameter);
right_len = left>right?left+1:right+1; //右子树最大深度取最大值 + 1
}
else{
right_len = 0;
}
diameter = diameter > left_len + right_len ? diameter : left_len + right_len;
//最大路径长度为左子树最大深度和右子树最大深度的和与原值取较大值
}