题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。(相当于是把68-I的搜索二叉树换成了普通的二叉树)
解题思路
来源:作者:jyd
链接:https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof/solution/mian-shi-ti-68-ii-er-cha-shu-de-zui-jin-gong-gon-7/
来源:力扣(LeetCode)
换成普通二叉树后,没有了搜索二叉树的定义条件,因此整体做法发生了改变,采用递归的方式去找p,q节点。
递归解析:
- 终止条件:
- 当越过叶节点,则直接返回 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 。具体可分为两种情况:
(1)p,q 其中一个在 root 的 右子树 中,此时 right 指向 p (假设为 p );
(2)p,q 两节点都在 root 的 右子树 中,此时的 right 指向 最近公共祖先节点 ;
- 当 left 不为空 , right 为空 :与情况 3. 同理;
复杂度分析:
时间复杂度 O(N) : 其中 N 为二叉树节点数;最差情况下,需要递归遍历树的所有节点。
空间复杂度 O(N) : 最差情况下,递归深度达到 N ,系统使用 O(N) 大小的额外空间。
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) return right;
if(right == null) return left;
return root;
}
}
改进点
这里主要是加入了一个标志位提前判断下是否已经找到最近公共祖先,比如早就找到了的话,其实没必要在进行后续的一大堆递归了,比如在root的左子树里面找到了,那么就没必要再去递归右子树,直接return。
/**
* 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) {
boolean flagFind=false;
if(flagFind){ //如果已经找到了最近公共祖先,则之后的操作之前全部return,这样可以减少点时间。
return null;
}
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){ return right; }
else if(right==null){ return left; }
else{
flagFind=true;
return root;
}
}
}