1 题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
示例1:
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出: 3
解释: 节点 5 和节点 1 的最近公共祖先是节点 3。
示例2:
输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出: 5
解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。
说明:
所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。
2 解题思路
祖先的定义: 若节点p在节点root的左(右)子树中,或p=root,则称root是p的祖先。
最近公共祖先的定义: 设节点root为节点p,q的某公共祖先,若其左子节点root.left和右子节点root.right都不是p,q的公共祖先,则称root是“最近的公共祖先”;
根据以上定义,若root是p,q的最近公共祖先,则只可能为以下情况之一:
- p和q在root的子树中,且分列root的异侧(即分别在左、右子树中);
- p=root,且q在root的左或右子树中;
- q=root,且p在root的左或右子树中;
考虑通过递归对二叉树进行后序遍历,当遇到节点p或q时返回。从底至顶回溯,当节点p,q在节点root的异侧时,节点root即为最近公共祖先,则向上返回root。
递归解析:
- 终止条件:
- 当越过叶节点,则直接返回null;
- 当root等于p,q,则直接返回root;
- 递推工作:
- 开启递归左子节点,返回值记为left;
- 开启递归右子节点,返回值记为right;
- 返回值:根据left和right,可展开为四种情况;
- 当left和right同时为空:说明root的左/右子树中都不包含p,q,返回null;
- 当left和right同时不为空:说明p,q分列在root的异侧(分别在左/右子树),因此root为最近公共祖先,返回root;
- 当left为空,right不为空:p,q都不在root的左子树中,直接返回right。具体可分为两种情况:
- p,q其中一个在root的右子树中,此时right指向p(假设为p);
- p,q两节点都在root的右子树中,此时的right指向最近公共祖先节点;
- 当left不为空,right为空:与情况3.同理;
/**
* 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 || root == p || root == q) return root;
TreeNode left = lowestCommonAncestor(root.left,p,q);
TreeNode right = lowestCommonAncestor(root.right,p,q);
if (left == null && right == null) return null;
else if (left == null) return right;
else if (right == null) return left;
else return root;
}
}
复杂度分析:
- 时间复杂度 O(N) : 其中 N 为二叉树节点数;最差情况下,需要递归遍历树的所有节点。
- 空间复杂度 O(N) : 最差情况下,递归深度达到 N ,系统使用 O(N) 大小的额外空间。