目录链接:
力扣编程题-解法汇总_分享+记录-CSDN博客
GitHub同步刷题项目:
https://github.com/September26/java-algorithms
原题链接:. - 力扣(LeetCode)
描述:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
示例 1:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8 输出: 6 解释: 节点2
和节点8
的最近公共祖先是6。
示例 2:
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4 输出: 2 解释: 节点2
和节点4
的最近公共祖先是2
, 因为根据定义最近公共祖先节点可以为节点本身。
说明:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉搜索树中。
解题思路:
我的思路是构建一个递归方法,传入值为当前遍历的节点node,搜索的节点searchNode,以及根结点到搜索节点的集合list,返回值为当前节点及其子节点是否包含搜索节点。
则递归方法中分为几种情况:
1.当前节点就是搜索的节点,则把当前节点加入到集合中,并返回true。
2.如果不是,则搜索其左子节点,如果包含,则加入到集合list中,并返回true。
3.右子节点也是相同的操作。
4.都没有,则返回false。
最终会得到两个list,分别代表从根节点到两个被搜索节点的所有节点。
因为都从根节点开始的,所以如果拥有相同的最近祖先,则其在list中的位置一定是一样的。所以寻找两个list中,最后一个相同的节点就是最近祖先。
当然这是二叉搜索树,实际上还可以优化,比如只寻找左或者右子树。
代码:
class Solution {
public:
TreeNode *lowestCommonAncestor(TreeNode *root, TreeNode *p, TreeNode *q)
{
vector<TreeNode *> pList;
search(root, p, pList);
vector<TreeNode *> qList;
search(root, q, qList);
int minSize = min(pList.size(), qList.size());
for (int i = minSize - 1; i >= 0; i--)
{
if (pList[i] == qList[i])
{
return pList[i];
}
}
return nullptr;
}
bool search(TreeNode *root, TreeNode *node, vector<TreeNode *> &list)
{
if (root == nullptr)
{
return false;
}
if (root == node)
{
list.push_back(root);
return true;
}
if (search(root->left, node, list))
{
list.insert(list.begin(), root);
return true;
}
if (search(root->right, node, list))
{
list.insert(list.begin(), root);
return true;
}
return false;
}
};