给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
一次遍历
/**
* 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||p==root||q==root) return root;//先序遍历,一旦找到立刻返回
TreeNode l=lowestCommonAncestor(root.left, p, q);
TreeNode r=lowestCommonAncestor(root.right, p, q);
return l==null?r:r==null?l:root;
//l==null r==null return null (p=100, q=100)
//l==null r!=null return r (p=1, q=8)
//l!=null r==null return l (p=5, q=4)
//l!=null r!=null return root (p=5, q=8)
}
}
两次遍历
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) {//将所有p的父节点都存入集合
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;
}
}