二叉搜索树的最近公共祖先
解法一:迭代法
Java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
while(root != null) {
if(root.val > p.val && root.val > q.val) // p,q都在 root的左子树中
root = root.left; // 遍历至 root的左子节点
else if(root.val < p.val && root.val < q.val) // p,q都在 root的右子树中
root = root.right; // 遍历至 root的右子节点
else // p,q在 root的两侧,则 root即为 p,q 的最近公共祖先
break;
}
return root;
}
}
Python
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
while root: # 注意 root的值也在不断更新的,所以是不断向下遍历的过程
if p.val < root.val and q.val < root.val: # 如果 p和 q都位于左子树中,则去左子树中继续找
root = root.left
elif p.val > root.val and q.val > root.val: # 如果 p和 q都位于右子树中,则去右子树中继续找
root = root.right
else: # 如果 p和 q一个位于左子树,一个位于右子树中,则返回根节点
return root
解法二:递归
Java
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root.val > p.val && root.val > q.val) // 当 p, q 都在 root的左子树中,则递归至 root.left
return lowestCommonAncestor(root.left, p, q);
if(root.val < p.val && root.val < q.val) // 当 p, q 都在 root的右子树中,则递归至 root.right
return lowestCommonAncestor(root.right, p, q);
return root; // p, q 分列 root的异侧(即分别在左、右子树中),则 root即为最近公共祖先
}
}
Python
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
if p.val < root.val and q.val < root.val: # 如果 p和 q都位于左子树中,则去左子树中继续找
return self.lowestCommonAncestor(root.left, p, q)
if p.val > root.val and q.val > root.val: # 如果 p和 q都位于右子树中,则去右子树中继续找
return self.lowestCommonAncestor(root.right, p, q)
return root # 如果 p和 q一个位于左子树,一个位于右子树中,则返回根节点
二叉树的最近公共祖先
Java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
// 递归终止条件
if(root == null) return null; // 如果树为空,直接返回null
if(root == p || root == q) return root; // 如果 p和q中有等于 root的,那么它们的最近公共祖先即为root
// 递推左子树和右子树
TreeNode left = lowestCommonAncestor(root.left, p, q); // 递归遍历左子树,只要在左子树中找到了p或q,则先找到谁就返回谁
TreeNode right = lowestCommonAncestor(root.right, p, q); // 递归遍历右子树,只要在右子树中找到了p或q,则先找到谁就返回谁
if(left == null) return right; // 如果在左子树中,p和 q都找不到,则 p和 q一定都在右子树中,先遍历到的那个就是最近公共祖先(一个节点也可以是它自己的祖先)
else if(right == null) return left; // 否则,如果 left不为空,在左子树中有找到节点(p或q),这时候要再判断一下右子树中的情况,如果在右子树中,p和q都找不到,则 p和q一定都在左子树中
else return root; //否则,当 left和 right均不为空时,说明 p、q节点分别在 root异侧, 最近公共祖先即为 root
}
}
Python
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
if not root: # 如果树为空,返回none;同时也作为递归的终止条件之一
return None
if root == p or root == q: # 如果 p和q中有等于root的,那么它们的最近公共祖先即为root;同时也作为递归的终止条件之一
return root
left = self.lowestCommonAncestor(root.left, p, q) # 递归遍历左子树,只要在左子树中找到了p或q,则先找到谁就返回谁
right = self.lowestCommonAncestor(root.right, p, q) # 递归遍历右子树,只要在右子树中找到了p或q,则先找到谁就返回谁
if not left: # 如果在左子树中,p和 q都找不到,则 p和 q一定都在右子树中,先遍历到的那个就是最近公共祖先,因为一个节点也可以是它自己的祖先
return right
elif not right: # 如果在左子树中有找到节点(p或q),这时候要再判断一下右子树中的情况,如果在右子树中,p和q都找不到,则 p和q一定都在左子树中
return left
else: # 如果在左子树和右子树中都有找到节点,则 p和 q一个存在于左子树,一个存在于右子树,最近公共祖先只能为root
return root