二叉搜索树的最近公共祖先(java)

题目描述:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
二叉搜索树的最近公共祖先(java)_第1张图片
二叉搜索树的最近公共祖先(java)_第2张图片
我们来看一下二叉搜索树的特征:对于一个结点而言,它的左孩子的值一定比它小,右孩子的值一定比它大。

故根据这个特点我们可以将题目整理成三种情况:

  1. 当给定的p q 结点的 val 都比 root 的值小,则说明应该在根结点的左子树中找最近公共祖先。(因为根结点的左孩子结点,不一定是 p q 的最近公共祖先,故仍需向下遍历)
  2. 当给定的p q 结点的 val 都比 root 的值大,则说明应该在根结点的右子树中找最近公共祖先。(右子树也许向下遍历)
  3. 当 p q 结点分在当下根结点的左右两边时,即可说明该结点就是最近公共祖先。

代码如下

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        //根结点的值
        int parentVal = root.val;
        //p结点的值
        int pVal = p.val;
        //q结点的值
        int qVal = q.val;

        if (pVal > parentVal && qVal > parentVal) {
            //pq结点都在根结点的右子树上,则从根结点的右孩子往下遍历
            return lowestCommonAncestor(root.right, p, q);
        } else if (pVal < parentVal && qVal < parentVal) {
            //pq结点都在根结点的左子树上,则从根结点的左孩子往下遍历
            return lowestCommonAncestor(root.left, p, q);
        } else {
            //如果他们两个没在一边,则说明找到了最近结点值
            return root;
        }
    }
}

你可能感兴趣的:(练习)