JavaScript实现LeetCode第235题:二叉搜索树的最近公共祖先
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
6
/ \
2 8
/ \ / \
0 4 7 9
/ \
3 5
示例 1:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。
示例 2:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
输出: 2
解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。
说明:
二叉搜索树的性质:
/**
* 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) {
let parentVal = root.val;
let pVal = p.val;
let qVal = q.val;
// 如果p、q均大于root,则应该由root的右子树返回p、q的最近公共祖先
if(pVal > parentVal && qVal > parentVal) {
return lowestCommonAncestor(root.right, p, q)
} else if(pVal < parentVal && qVal < parentVal) {
// 如果p、q均小于root,则应该由root的左子树返回p、q的最近公共祖先;
return lowestCommonAncestor(root.left, p, q)
} else {
// p、q既不同时大于root,也不同时小于root,则root就是p、q的最近公共祖先,返回root
return root;
}
};
用迭代的方式替代了递归来遍历整棵树。由于我们不需要回溯来找到 LCA 节点,所以我们是完全可以不利用栈或者是递归的。
/**
* 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) {
let pVal = p.val;
let qVal = q.val;
let node = root;
while (node !== null) {
let parentVal = node.val;
if(pVal > parentVal && qVal > parentVal) {
node = node.right;
} else if(pVal < parentVal && qVal < parentVal) {
node = node.left;
} else {
return node;
}
}
return null;
};