Leetcode刷题(第236题)——二叉树的最近公共祖先
一、题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。 百度百科中最近公共祖先的定义为:“对于有根树 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 。
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4 输出:5 解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身
输入:root = [1,2], p = 1, q = 2 输出:1
三、思路 我们可以使用递归算法,将根节点的左节点和右节点分别使用递归,然后设置递归的终止条件,如果root节点为null,或者root = p, 或者root = q,则直接返回root。我们进行判断,如果左节点递归存在返回值,并且右节点也存在,则直接返回root即可。如果左节点存在返回值,右节点递归不存在,则直接返回左节点。反之返回右节点。 四、代码
/** * 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) { if(root === null || root === p || root === q) { return root } let left = lowestCommonAncestor(root.left, p, q) let right = lowestCommonAncestor(root.right, p, q) if(left && right) { return root } return left ? left : right }; //3 5 1 // 5 5 1 // 1 5 1
五、总结