力扣235 二叉搜索树中的最近公共祖先

题目链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

例如,给定如下二叉搜索树:  root = [6,2,8,0,4,7,9,null,null,3,5]

 

示例 1:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6 
解释: 节点 2 和节点 8 的最近公共祖先是 6。
示例 2:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。
 

说明:

  • 所有节点的值都是唯一的。
  • p、q 为不同节点且均存在于给定的二叉搜索树中。

 

        其实很简单,注意要求是返回结点,而不是值。因为是二叉搜索树,所以一个结点的左子树所有结点的值都小于等于该点的值,左子树所有结点的值都大于等于该点的值,原题目说明中第一条“所有节点的值唯一”那么上述不存在等于的情况。也就是说,对于根节点,如果任一结点的值小于其值,那么该结点位于其左子树中,大于则位于右子树中。如果两个结点分别位于根节点的左、右子树中(即根节点的值介于两个结点的值中间),显而易见他们的最近公共祖先是根节点;否则两个结点同位于根节点的左、右子树中(即两个结点都比根节点值大或小),那么两个结点的最近公共祖先也位于左、右子树中,并且最近公共祖先的值介于两个结点的值之间。

        算法可通过递归/迭代来完成,输入为子树的根结点root,两个结点p和q,设vallow为p和q两个结点值的最小值,valhigh为最大值,则:

  1. 如果root的值大于等于vallow但小于等于valhigh的值,说明介于两者值之间,两个结点分别位于其左右子树,那么root即他们的最近公共祖先;
  2. 否则如果root的值大于valhigh的值,说明两个结点的值都比root的值小,即都位于左子树中,则进入root的左子树中寻找;
  3. 否则root的值小于vallow的值,说明两个结点的值都比root的值大,即都位于右子树中,则进入root的右子树中寻找。

       情况1中两边取等号的原因为最近公共祖先节点可以为节点本身,所以如果存在相等的情况,说明两个结点中有一个结点为根节点。

递归方法:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        int vallow=min(p->val,q->val);
        int valhigh=max(p->val,q->val);
        if (vallow<=root->val && valhigh>=root->val)
            return root;
        else if(valhigh<root->val)
            return lowestCommonAncestor(root->left,p,q);
        else
            return lowestCommonAncestor(root->right,p,q);
        }
};

迭代方法:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        TreeNode* r=root;
        int vallow=min(p->val,q->val);
        int valhigh=max(p->val,q->val);
        while(r){
            if (vallow<=r->val && valhigh>=r->val)
                break;//其实这行可以直接return r的,但是不通过
            else if(valhigh<r->val)
                r=r->left;
            else
                r=r->right;
        }
        return r;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值