从二叉树的节点A出发,可以向上或者向下走,但沿途的节点只能经过一次,当到达节点B时,路径上的节点数叫作A到B的距离。对于给定的一棵二叉树,求整棵树上节点间的最大距离。
给定一个二叉树的头结点root,请返回最大距离。保证点数大于等于2小于等于500.
思路:修改后序遍历
对于给定的节点root,最远距离只可能来自以下三种情况:
1、左子树的最远距离
2、右子树的最远距离
3、左子树上距离root最远长度和右子树上距离root最远长度之和加1;
递归遍历即可。每个递归返回两个值,一个是距离root的最远距离,另一个是以root为头节点的树上最远距离,使用动态数组或者vector,不能直接返回数组指针,每个局部函数执行完内存就释放掉了。
计算结点root所在子树的最长距离,需要已知:左子树的最长距离left[0],左子树的高度left[1];右子树的最长距离right[0],右子树的高度right[1].
然后比较max(left[0],right[0],(left[1]+right[1]+1)),最大值就是这棵二叉树的最大距离,即对于每个子树,需要求出它的最大距离和最大高度。
可以使用递归来实现,求出这个根结点的最大距离a[0]和最大高度a[1]并返回这2个数值。其中a[0]= max(left[0],right[0],(left[1]+right[1]+1));
而由于要返回这棵子树的高度,如何求子树的高度呢?二叉树的高度就是它的2个子树高度中的较大值再加上1,即a[1]=max(left[1], right[1])+1;
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class LongestDistance {
public:
int findLongest(TreeNode* root) {
if(!root) return 0;
vector<int> res=getDist(root);//调用递归函数来求得最大距离a[0]和高度a[1]
return res[0];
}
vector<int> getDist(TreeNode* root){
vector<int> a(2,0);
if(!root) return a;
vector<int> left=getDist(root->left);//左子树的结果
vector<int> right=getDist(root->right);//右子树的结果
a[1]=max(left[1],right[1])+1;//最远距离
a[0]=max(max(left[0],right[0]),left[1]+right[1]+1);//最远长度
return a;
}
};
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class LongestDistance {
public:
int findLongest(TreeNode* root) {
int dist=0;
getMaxDist(root,dist);
return dist;
}
int getMaxDist(TreeNode* node,int& dist){
if(!node)return 0;
int leftDist=getMaxDist(node->left,dist);
int rightDist=getMaxDist(node->right,dist);
dist=max(leftDist+rightDist+1,dist);
return max(leftDist,rightDist)+1;
}
};