二叉树的最近公共祖先
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
思路:
- p、q位于一个根结点的两个子树,或者位于一个子树。
- 采用迭代的方法,搞清楚迭代结束的条件。
代码:
方法一:递归实现
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(!root ||root==p || root==q) //迭代结束条件
return root;
TreeNode* left=lowestCommonAncestor(root->left,p,q);
TreeNode* right=lowestCommonAncestor(root->right,p,q);
if( left && right ) return root; //位于2个子树
return left?left:right; //位于1个子树
}
};
方法二:非递归实现
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
unordered_map<TreeNode*,TreeNode*> parent;
queue<TreeNode*> que;
if(root!=nullptr)
{
que.push(root);
parent[root]=nullptr;
}
//(1)BFS,记录每个点的父节点,只需要记录到p,q,后面的不用管
TreeNode* cur;
while(parent.find(p)==parent.end() || parent.find(q)==parent.end() ){
cur=que.front();
que.pop();
if(cur->left)
{
parent[cur->left]=cur;
que.push(cur->left);
}
if(cur->right)
{
parent[cur->right]=cur;
que.push(cur->right);
}
}
//(2)记录p的祖先,包括其自身
unordered_set<TreeNode*> ancestor;
while(p!=nullptr)
{
ancestor.insert(p);
p=parent[p];
}
//(3)寻找公共祖先
while(ancestor.find(q)==ancestor.end())
{
q=parent[q];
}
return q;
}
结果:
参考链接:
[1] 重剑DS:236. 二叉树的最近公共祖先(C++)