有一棵二叉树,树上每个点标有权值,权值各不相同,请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为1,一个节点经过多少条边到达另一个节点为这两个节点之间的距离。
给定二叉树的根节点root,请返回所求距离。
解法:
先利用getMaxandMin函数找到叶子结点中最大值和最小值对应的结点,然后利用tree函数,递归返回一个pair<int,int>,第一个值是标志位,为1表示当前节点是最大值或最小值的祖先,第二个值是当前节点到最大值或最小值路径上经过的结点数。若左右子树的tree函数的返回值中first的值都为1时,表示当前节点是最大值和最小值的最近公共祖先,更新ans=pl.second+pr.second;
class Solution {
public:
int ans=0;
int maxnum=-20000000;
int minnum=20000000;
TreeNode* maxp=NULL;
TreeNode* minp=NULL;
void getMaxandMin(TreeNode* root){
if(!root)
return;
if(root->left==NULL&&root->right==NULL){
if(root->val>maxnum){
maxp=root;
maxnum=root->val;
}
if(root->val<minnum){
minp=root;
minnum=root->val;
}
}
getMaxandMin(root->left);
getMaxandMin(root->right);
}
pair<int,int> tree(TreeNode* root,TreeNode* p, TreeNode* q){
if(!root){
return make_pair(0,0);
}
pair<int,int> pl=tree(root->left,p,q);
pair<int,int> pr=tree(root->right,p,q);
if(root->val==p->val||root->val==q->val)
return make_pair(1,1);
if(pl.first==1&&pr.first==1){
ans=pl.second+pr.second;
return make_pair(0,0);
}
else if(pl.first==1&&pr.first==0){
return make_pair(1,pl.second+1);
}
else if(pl.first==0&&pr.first==1){
return make_pair(1,pr.second+1);
}
return make_pair(0,0);
}
int* minDistance(TreeNode* root) {
getMaxandMin(root);
tree(root,maxp,minp);
return ans;
}
};