【面试题】树中两个节点的最近公共祖先

二叉搜索树的最近公共祖先

题解参考

解法一：迭代法

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