剑指No.68_1_二叉搜索树的最近公共祖先
- 题目:给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例:输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。
- MySolution
创建两个列表,分别保存遍历到q和p所经过的节点,最后再从列表长度较长的那个从后往前与另一个列表比较,直到找到相同的节点。(因为是二叉搜索树,所以每遍历过一个节点深度就加1)
public TreeNode lowestCommonAncestorWay(TreeNode root, TreeNode p, TreeNode q){
List<TreeNode> pList = new ArrayList<>();
List<TreeNode> qList = new ArrayList<>();
TreeNode temp = root;
while (root != null){
pList.add(root);
if (root.val == p.val){
break;
}
if (p.val > root.val){
root = root.right;
}else {
root = root.left;
}
}
root = temp;
while (root != null){
qList.add(root);
if (root.val == q.val)
break;
if (q.val > root.val)
root = root.right;
else
root = root.left;
}
int pLength = pList.size() - 1;
int qLength = qList.size() - 1;
while (pLength >= 0 && qLength >= 0){
if (pLength > qLength){
if (pList.get(--pLength) == qList.get(qLength))
return qList.get(qLength);
}else if (pLength < qLength){
if (qList.get(--qLength) == pList.get(pLength))
return pList.get(pLength);
}else{
if (qList.get(qLength) == pList.get(pLength))
return qList.get(qLength);
qLength--;
pLength--;
}
}
return null;
}
- OfficialSolution
按照二叉搜索树的特性和最近公共祖先的性质,可以发现从根节点开始遍历,p和q第一次分开处就是最近公共祖先。
public TreeNode lowestCommonAncestorWay(TreeNode root, TreeNode p, TreeNode q){
while (root != null){
if (p.val > root.val && q.val > root.val)
root = root.right;
else if (p.val < root.val && q.val < root.val)
root = root.left;
else
break;
}
return root;
}