剑指 68-I. 二叉搜索树的最近公共祖先

1. 题目描述

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-lcof

2. 题解

1、方法1：两次遍历

思路：从根节点遍历二叉树，记录从root到节点p, q的路径，然后比较这两条路径，返回两条路径中最后一个相同的节点。

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        TreeNode node = root;
        List<TreeNode> path1 = getPath(root, p);
        List<TreeNode> path2 = getPath(root, q);
        //找公共部分的最后一个节点
        int i = 0, j = 0;
        while(i < path1.size() && j < path2.size()){
            if(path1.get(i) == path2.get(j)){
                i++;
                j++;
            }else{
                break;
            }
            node = path1.get(i-1);
        }
        return node;
    }

    // 遍历二叉树并存储从root到target的路径
    public List<TreeNode> getPath(TreeNode root, TreeNode target){
        List<TreeNode> path = new ArrayList<TreeNode>();
        TreeNode node = root;
        while(node != target){
            path.add(node);
            if(node.val > target.val){
                node = node.left;
            }else{
                node = node.right;
            }
        }
        // 注意这里节点自身也可做祖先
        path.add(node);
        return path;
    }
}

时间复杂度：O(n)，n为二叉树的节点个数，时间复杂度和p, q节点在树中的深度有关，最坏情况是O(n)。
空间复杂度：O(n)， 存储路径所需的空间。

2、方法2：一次遍历

思路：
当前节点和p, q 共有3种关系：
（1）当前节点大于p, q 的值，那么转向左子树寻找；
（2）当前节点小于p, q 的值，转向右子树寻找；
（3）当前节点是分岔点，p, q 要么分别在当前节点的左右子树，要么等于当前节点，此时当前节点就是要找的最近公共祖先。

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

时间复杂度：O(n)，n为二叉树的节点个数，时间复杂度和p, q节点在树中的深度有关，最坏情况是O(n)。
空间复杂度：O(1)， 省去了存储路径所需的空间。