235. (LCA) Lowest Common Ancestor of a Binary Search Tree

ps. 這題用 leetcode 236 也可以解決, 速度還差不多...但因為這題是 BST, 所以作法更簡單
/*
235. Lowest Common Ancestor of a Binary Search Tree
level: easy
https://leetcode.com/problems/lowest-common-ancestor-of-a-binary-search-tree/
https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree/
BST LCA
如果只看祖先, 所以 3, 5 祖先有 6, 2, 4
但最矮的 lca, 一定會讓 3, 5 在左右兩側
其他可以發現, 3,5 都在 某一側(6, 2 的某一側
因為 bst 有左小右大的特性 (左子樹所有節點皆小於root, 右子樹所有節點皆大於root
1. 所以pq同時比root 大時, 代表pq在右邊, 所以要往右走
2. 所以pq同時比root 小時, 代表pq在左邊, 所以要往左走
3. 其他狀況, 代表找到了, p q在左右兩側了
Input: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
Output: 6
Explanation: The LCA of nodes 2 and 8 is 6.
LCS:
判斷方式: 找到 LCS的方式 p, q 不在此 LCS 同一側
//說明1
if (pVal > parentVal && qVal > parentVal) {
// 往右走~~
return lowestCommonAncestor(root.right, p, q);
} else if (pVal < parentVal && qVal < parentVal) { //說明2
// 往左走~~
return lowestCommonAncestor(root.left, p, q);
} else {//說明3
//找到了 We have found the split point, i.e. the LCA node.
return root;
}
Time Complexity: O(N),
where N is the number of nodes in the BST. In the worst case we might
be visiting all the nodes of the BST.
Space Complexity: O(N). This is because the maximum amount of
space utilized by the recursion stack would be N since the height of a
skewed BST could be N
*/
public class LowestCommonAncestorBST {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
int parentVal = root.val;
int pVal = p.val;
int qVal = q.val;
if (pVal > parentVal && qVal > parentVal) {
return lowestCommonAncestor(root.right, p, q);
} else if (pVal < parentVal && qVal < parentVal) {
return lowestCommonAncestor(root.left, p, q);
} else { //terminator
return root;
}
}
}Last updated
Was this helpful?