题目:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
实例一:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。
实例二:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。
示例三:
输入:root = [1,2], p = 1, q = 2 输出:1
提示:
树中节点数目在范围 [2, 105] 内。
-109 <= Node.val <= 109
所有 Node.val 互不相同 。
p != q
p 和 q 均存在于给定的二叉树中。
思路:
首先我们先来看一眼题目,题目只有一个要求,就是找出二叉树中两个指定节点的最近公共祖先,随后给出了最近公共祖先的概念“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)”。
这道题我们可以使用哈希表先将二叉树中每个节点的父节点都存起来,又因为提示当中给出了所有 Node.val 互不相同,所以我们可以用每个节点的val值去当作哈希表的key值,在存完每个节点的父节点之后,我们先从给定的两个节点其中的一个不断向上去走(通过哈希表去走父节点)并将路过的每个节点都记录下来(这里我们使用哈希表去存)直到第一个节点走到头。随后让第二个节点也网上走,只要在标记哈希表查到这个点被标记过,即这个点就是他俩的公共祖节点,否则就是没有。
代码:
/**
* 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:
unordered_map<int, TreeNode*> father;
unordered_map<int, bool> flags;
void dfs(TreeNode* root){
if(root->left){
father[root->left->val] = root;
dfs(root->left);
}
if(root->right){
father[root->right->val] = root;
dfs(root->right);
}
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
dfs(root);
while(p){
flags[p->val] = true;
p = father[p->val];
}
while(q){
if(flags[q->val]) return q;
q = father[q->val];
}
return nullptr;
}
};
unordered_map<int, TreeNode*> father;
unordered_map<int, bool> flags;
father就是我们用来存每个节点的父节点的哈希表
flags就是我们用来存第一个节点在向上走的过程当中路过的点的哈希表
void dfs(TreeNode* root){
if(root->left){
father[root->left->val] = root;
dfs(root->left);
}
if(root->right){
father[root->right->val] = root;
dfs(root->right);
}
}
这是为了完成记录父节点而写的一个递归函数,就是一个普通的递归函数
dfs(root);
while(p){
flags[p->val] = true;
p = father[p->val];
}
while(q){
if(flags[q->val]) return q;
q = father[q->val];
}
接下来到了主函数的部分,我们先完成了每个节点的父节点的记录,随即开始了第一个节点p向上走的过程,每一个走过的节点标记为true,下面就是第二个节点q向上走的过程,一但发现某一个点标记为true,即这个点就是两个节点的共同的祖先节点,直接返回即可,反之到函数的最后,返回nullptr。