leetcode之二叉树七: 二叉搜索树的最近公共祖先_7-1 二叉搜索树的最近公共祖先测试点-CSDN博客

本文链接：https://blog.csdn.net/qq_32523587/article/details/116888193

原题: 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 为不同节点且均存在于给定的二叉搜索树中。

二、解题

package com.leetcode.test.tree;

public class Solution7 {
    public static void main(String[] args) {
        TreeNode a1 = new TreeNode(6);
        TreeNode a2 = new TreeNode(2);
        TreeNode a3 = new TreeNode(8);
        TreeNode a4 = new TreeNode(0);
        TreeNode a5 = new TreeNode(4);
        TreeNode a6 = new TreeNode(7);
        TreeNode a7 = new TreeNode(9);
        TreeNode a8 = new TreeNode(3);
        TreeNode a9 = new TreeNode(5);

        a1.left = a2;
        a1.right = a3;
        a2.left = a4;
        a2.right = a5;
        a3.left = a6;
        a3.right = a7;
        a5.left = a8;
        a5.right = a9;

        System.out.println(lowestCommonAncestor(a1, a2, a3).val);

    }

    //因为是二叉搜索树,
    //如果p、q都小于当前节点，则在左子树
    //如果p、q都大于当前节点，则在右子树
    //如果p、q一个大于当前节点，一个小于当前节点，则一个在左，一个在右, 该节点即是最近公共祖先
    public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //如果p、q都小于当前节点，则在左子树
        if (p.val < root.val && q.val < root.val) {
            return lowestCommonAncestor(root.left, p, q);
        }

        //如果p、q都大于当前节点，则在右子树
        if (p.val > root.val && q.val > root.val) {
            return lowestCommonAncestor(root.right, p, q);
        }

        //如果p、q一个大于当前节点，一个小于当前节点，则一个在左，一个在右, 该节点即是最近公共祖先
        if ((p.val < root.val && q.val > root.val) || (p.val > root.val && q.val < root.val)) {
            return root;
        }

        return root;
    }
}