BST有一个特性,就是某个root结点的值大于左子树中所有结点的值,小于所有右子数结点的值。所以如果一个结点是另外两个节点的共同祖先,那么这个结点的值必然是在另外两个值之间,可以根据这个特性写出算法。另外一点常用的思维就是在遇到二叉树的问题的时候,常常需要用递归的思维解决问题。另外需要注意的一点是,由于题目中给出的是某个节点的ancestor可以是自身,所以下面代码中的else中其实隐含了这种情况。代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
// Method One: Recursive
public class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root.val > p.val && root.val > q.val){
return lowestCommonAncestor(root.left, p, q);
}else if(root.val < p.val && root.val < q.val){
return lowestCommonAncestor(root.right, p, q);
}else{
return root;
}
}
}
// Method Two: Non-recursive
public class Solution{
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q){
while(true){
if(root.val > p.val && root.val > q.val) root = root.left;
else if(root.val < p.val && root.val < q.val) root = root.right;
else return root;
}
}
}