leetcode中等难度题 -- 34.二叉树的最近公共祖先

问题描述

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

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3
解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

解题思路：递归

1. 定义递归函数traveTree(root, p, q)，有以下4种情况：

① 当root === null || root === p || root === q时，表示当前树为空 || root为p, q中任意节点，可以直接返回root

②递归左右子树，得到left, right，如果left和right都不为空，说明p和q存在于root的左右子树中，返回root

③如果left为空，说明p, q在右子树中，返回right

④如果right为空，说明p, q在左子树中，返回left

2. 调用递归函数traveTree(root, p, q)，得到最终结果

代码实现（js）

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {TreeNode}
 */
var lowestCommonAncestor = function(root, p, q) {
    // 1.确定递归函数
    const traveTree = (root, p, q) => {
        // 1.1 终止条件：root === null || root === p || root === q  => return root
        if(root === null || root === p || root === q){
            return root
        }
        // 1.2 递归左右子树
        let left = traveTree(root.left, p, q)
        let right = traveTree(root.right, p, q)
        // 1.3 如果左右子树都不为空
        if(left !== null && right !== null){
            return root
        }
        if(left === null){
            // p, q都在右子树上
            return right
        }
        if(right === null){
            // p, q都在左子树上
            return left
        }
    }
    // 递归开始
    return traveTree(root, p, q)
};