LCA of a Binary Tree

236. Lowest Common Ancestor of a Binary Tree

 

 

    /**
     * 基础版
     * 给定p，q都是在树中
     * 有两种情况：
     *   1. p和q分布在LCA的两侧
     *   2. p和q在同一侧（p是q的祖先 或者是 q是p的祖先）
     * 先考虑第一种情况，例如：
     *                  1
     *                 / \
     *                2   4
     * 2和4的LCA是1，向下递归返回时，2返回2，4返回4，1接受到的left和right的返回值都存在值，那么就返回自己
     * 也就是说=> 只要找到目标节点，就往上返回该节点
     * ---------------------------------
     * 第二种情况，例如：
     *              1
     *             /
     *            2
     *             \
     *              3
     * 2和3的LCA是2，2接收到的是3，而自己也是目标节点，那么就返回自己
     * --------------注意1----------------
     * 不用考虑在1处，只有一个返回值，如何判断是否有效。因为：两个节点必定在树中，如果root接收到一个值，那么就表示root之下就有p和q的LCA
     * 如果接收到两个值，那么自己就是LCA
     * --------------注意2----------------
     * Corner case: p就是root 或者 q就是root
     */
    public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // base case:
        // 如果当前结点(root)是p或者是q，就返回自己
        if (root == null || root == p || root == q) {
            return root;
        }

        TreeNode leftNode = lowestCommonAncestor(root.left, p, q);
        TreeNode rightNode = lowestCommonAncestor(root.right, p, q);

        if (leftNode != null && rightNode != null) {    // 接收到两个值，返回root
            return root;
        } else if (leftNode != null && rightNode == null) { // 接收到一个值，就返回该node
            return leftNode;
        } else {
            return rightNode;
        }
    }