给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
自我解(递归,时间复杂度为O(深度*结点个数))
/**
* 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){
int mid = dfs(root.left,p,q);
if(root.val == p.val||root.val == q.val)
break;
else if (mid == 1)
break;
else if(mid == 2)
root = root.left;
else
root = root.right;
}
return root;
}
public int dfs(TreeNode root , TreeNode p , TreeNode q){
if(root==null)
return 0;
int mid = 0;
if(root.val==p.val||root.val==q.val)
mid = 1;
int l = dfs(root.left,p,q);
int r = dfs(root.right,p,q);
return l+r+mid;
}
}
自我解(递归,时间复杂度为O(n))
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
TreeNode result;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
dfs(root,p,q);
return result;
}
public boolean dfs(TreeNode root , TreeNode p , TreeNode q){
if(root == null)
return false;
boolean l = dfs(root.left,p,q);
boolean r= dfs(root.right,p,q);
if((l&&r)||(root.val == p.val||root.val==q.val)&&(l||r))
result = root;
return l||r||root.val==p.val||root.val==q.val;
}
}
官方解(哈希表存储父节点,时间复杂度为O(n))
class Solution {
Map<Integer, TreeNode> parent = new HashMap<Integer, TreeNode>();
Set<Integer> visited = new HashSet<Integer>(); //存储两个结点已到达路径,如果重合,则表明是最近公共祖先
public void dfs(TreeNode root) {
if (root.left != null) {
parent.put(root.left.val, root); //存储父节点对应关系
dfs(root.left);
}
if (root.right != null) {
parent.put(root.right.val, root); //存储父节点对应关系
dfs(root.right);
}
}
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
dfs(root);
while (p != null) {
visited.add(p.val); //存储路径
p = parent.get(p.val);
}
while (q != null) {
if (visited.contains(q.val)) { //如果重合,则最近父节点
return q;
}
q = parent.get(q.val);
}
return null;
}
}