算法---LeetCode 235. 二叉搜索树的最近公共祖先(同剑指offer 68-1)

1. 题目

原题链接
剑指offer

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉搜索树中。

注意:本题与主站 235 题相同:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a
-binary-search-tree/
Related Topics 树

2. 题解

2.1 解法1: 迭代(一次遍历)

从根节点开始遍历;

  1. 如果当前节点的值大于 p 和 q 的值,说明 p 和 q 应该在当前节点的左子树,因此将当前节点移动到它的左子节点;
  2. 如果当前节点的值小于 p 和 q 的值,说明 p 和 q 应该在当前节点的右子树,因此将当前节点移动到它的右子节点;
  3. 如果当前节点的值不满足上述两条要求,那么说明当前节点就是「分岔点」。此时,p 和 q 要么在当前节点的不同的子树中,要么其中一个就是当前节点。
    相当于找分岔点
    class Solution {
        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
            while (root != null) {
                if (p.val < root.val && q.val < root.val) {
                    root = root.left;
                } else if (p.val > root.val && q.val > root.val) {
                    root = root.right;
                } else {
                    break;
                }
            }
            return root;
        }
    }

2.2 解法2: 递归

思路同解法1

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

2.3 解法3: 两次遍历

由于是二叉搜索树, 根节点大于左节点, 小于右节点,

  1. 第一次遍历查找 p 的位置, 第二次查找q, 同时记录路径
  2. 最后找到两条路径中最后一个相同的结点即为最近公共祖先
    class Solution {
        public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
            if (root == null) {
                return null;
            }
            List<TreeNode> pList = new ArrayList<>();
            List<TreeNode> qList = new ArrayList<>();
            TreeNode curr = root;
            while (curr != p) {
                pList.add(curr);
                if (p.val < curr.val) {
                    curr = curr.left;
                } else if (p.val > curr.val) {
                    curr = curr.right;
                }
            }
            pList.add(curr);
            curr = root;
            while (curr != q) {
                qList.add(curr);
                if (q.val < curr.val) {
                    curr = curr.left;
                } else if (q.val > curr.val) {
                    curr = curr.right;
                }
            }
            qList.add(curr);
            TreeNode ans = null;
            for (int i = 0; i < pList.size() && i < qList.size(); i++) {
                if (pList.get(i) == qList.get(i)) {
                    ans = pList.get(i);
                } else {
                    break;
                }
            }
            return ans;

        }
    }

参考:
官方题解
面试题68 - I. 二叉搜索树的最近公共祖先(迭代 / 递归,清晰图解)

上一篇:Leetcode刷题笔记之专题(1)链表求解 Python实现


下一篇:单链表算法