给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为: “对于有根树T的两个结点u、v,最近公共祖先表示一个结点x,满足x是u、v的祖先且x的深度尽可能大。”(一个节点也可以是它自己的祖先)
例如,给定二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
_______6______ / \ ___2__ ___8__ / \ / \ 0 _4 7 9 / \ 3 5
示例 1:
输入: root, p = 2, q = 8 输出: 6 解释: 节点2
和节点8
的最近公共祖先是6。
示例 2:
输入: root, p = 2, q = 4 输出: 2 解释: 节点2
和节点4
的最近公共祖先是2
, 因为根据定义最近公共祖先节点可以为指定节点自身。
思想:由于是基于二叉树搜索树进行查找,那么问题简化了很多。这里我用到了两个vector分别记录了两个结点的搜索路径,然后再遍历两个vector找出最后一个相同的结点,这里需要注意的是题目要求返回结点指针,那么在定义vector存储变量类型的时候需要注意了。细节见代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
vector<TreeNode*> left;
vector<TreeNode*> right;
TreeNode *pp = root;
TreeNode *qq = root;
while (pp != NULL)
{
left.push_back(pp);
if (p->val < pp->val)
{
pp = pp->left;
}
else if (p->val > pp->val)
{
pp = pp->right;
}
else
{
break;
}
}
while (qq != NULL)
{
right.push_back(qq);
if (q->val < qq->val)
{
qq = qq->left;
}
else if (q->val > qq->val)
{
qq = qq->right;
}
else
{
break;
}
}
int i = 0;
int j = 0;
TreeNode *temp;
while (i < left.size() && j < right.size())
{
if (left[i] == right[j])
{
temp = left[i];
}
i++;
j++;
}
return temp;
}
};