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

题目

思路

两次遍历

注意到题目中给出的是一棵「二叉搜索树」，因此我们可以快速地找出树中的某个节点以及从根节点到该节点的路径，例如我们需要找到节点 pp：

我们从根节点开始遍历；

如果当前节点就是 pp，那么成功地找到了节点；

如果当前节点的值大于 pp 的值，说明 pp 应该在当前节点的左子树，因此将当前节点移动到它的左子节点；

如果当前节点的值小于 pp 的值，说明 pp 应该在当前节点的右子树，因此将当前节点移动到它的右子节点。

对于节点 qq 同理。在寻找节点的过程中，我们可以顺便记录经过的节点，这样就得到了从根节点到被寻找节点的路径。

当我们分别得到了从根节点到 pp 和 qq 的路径之后，我们就可以很方便地找到它们的最近公共祖先了。显然，pp 和 qq 的最近公共祖先就是从根节点到它们路径上的「分岔点」，也就是最后一个相同的节点。因此，如果我们设从根节点到 pp 的路径为数组 \textit{path_p}[]path_p[]，从根节点到 qq 的路径为数组 \textit{path_q}[]path_q[]，那么只要找出最大的编号 ii，其满足

\textit{path_p}[i] = \textit{path_q}[i]
path_p[i]=path_q[i]

那么对应的节点就是「分岔点」，即 pp 和 qq 的最近公共祖先就是 \textit{path_p}[i]path_p[i]（或 \textit{path_q}[i]path_q[i]）

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

class Solution {
    ArrayList<TreeNode> path_p = new ArrayList<TreeNode>();
    ArrayList<TreeNode> path_q = new ArrayList<TreeNode>();
    ArrayList<TreeNode> path = new ArrayList<TreeNode>();
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        dfs(root, p.val);
        path_p = new ArrayList<TreeNode>(path);
        path.clear();
        dfs(root, q.val);
        TreeNode rst = null;
        path_q = new ArrayList<TreeNode>(path);
        for(int i = 0; i < Math.min(path_q.size(),path_p.size()); i++){
            if(path_p.get(i) == path_q.get(i)){
                rst = path_p.get(i);
            }
        }
        return rst;
    }

    public void dfs(TreeNode root, int tab){
        if (root.val == tab){
            path.add(root);
            return;
        }
        else if(tab < root.val){
            path.add(root);
            dfs(root.left, tab);
        }
        else {
            path.add(root);
            dfs(root.right, tab);
        }
    }
}