二叉树最大叶子结点到最小叶子结点的最短距离

有一棵二叉树,树上每个点标有权值,权值各不相同,请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为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;
        
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值