思路
有两种情况会成为公共祖先:
- 第一个分叉点
- 当前节点等于两个节点中的一个
代码
➥ JavaScript
/**
* 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 (p.val < root.val && q.val < root.val)
return lowestCommonAncestor(root.left, p, q)
else if (p.val > root.val && q.val > root.val)
return lowestCommonAncestor(root.right, p, q)
else
return root
};
进阶
还有一道相似的题目:二叉树的最近公共祖先。
相比于二叉搜索树,普通的二叉树是乱序的。这题的解题思路是,通过向下喊话,然后层层反馈来判断的。
比如根节点 root 问自己的左孩子 left,你是 p 吗?左孩子说不是。root 说,那你问问你的两个孩子是么。此时 left 变成了它的两个孩子的 root。
上面的步骤一层层的往下走直到找到 p 和 q,这时候开始往上反馈。
p 和 q 分别向自己的父节点说我就是你要找的节点,如果正好 p 和 q 的父节点相同,那么这就是最近的公共祖先。否则父节点再往上喊话:我找到了 q(或 p)。
一层层往上喊话,直到某个节点同时得到了 p 和 q 的消息,那么它就是最近公共祖先节点。
/**
* @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 // 已到底,或者找到了p或q
const left = lowestCommonAncestor(root.left, p, q) // 左子树的查找结果
const right = lowestCommonAncestor(root.right, p, q) // 右子树的查找结果
return left === null ? right : (right === null ? left : root) // 根据结果判断是找到了,还是继续往上喊话
};