235. 二叉搜索树的最近公共祖先

本文介绍了一种寻找二叉搜索树中两个节点的最近公共祖先的有效算法。通过中序遍历找到位于两节点值之间的节点,并进一步优化为直接比较当前节点值与目标节点值,快速定位到公共祖先。

235. 二叉搜索树的最近公共祖先

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        List<TreeNode> arr = new ArrayList<>();
        build(root,arr);
        int max = Math.max(p.val,q.val);
        int min = Math.min(p.val,q.val);
        TreeNode res = root;
        for(int i = 0; i < arr.size();i++) {
            if(arr.get(i).val >= min && arr.get(i).val <= max){
                if(find(arr.get(i),min) && find(arr.get(i),max)) {
                    res = arr.get(i);
                    break;
                }
            }else if (arr.get(i).val > max) i = Integer.MAX_VALUE;
        }
        return res;
    }
    public void build(TreeNode node, List<TreeNode> arr){
        if(node == null) return;
        build(node.left,arr);
        arr.add(node);
        build(node.right,arr);
    }
    public boolean find(TreeNode node, int val){
        if(node == null) return false;
        if(node.val == val) return true;
        return find(node.left,val) || find(node.right,val);
    }
}

中序遍历,公共祖先一定在p.val和q.val中间,中间每个子树都判断是否是其公共祖先即可。

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        TreeNode res = root;
        while(true){
            if(root.val > p.val && root.val > q.val){
                root = root.left;
            }else if(root.val < p.val && root.val < q.val){
                root = root.right;
            }else{
                res = root;
                break;
            }
        }
        return res;
    }
}

利用二叉搜索树的特性,当当前节点的值处于p和q之间,则为公共祖先

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值