1.题目如下:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例 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 。因为根据定义最近公共祖先节点可以为节点本身。
示例 3:
输入:root = [1,2], p = 1, q = 2
输出:1
提示:
树中节点数目在范围 [2, 105] 内。
-109 <= Node.val <= 109
所有 Node.val 互不相同 。
p != q p 和 q 均存在于给定的二叉树中。
2.代码如下:
/**
* 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 * res;
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
dps(root,p,q);
return res;
}
//该函数判断一个子树是否包含要查询的子节点
bool dps(TreeNode* root, TreeNode* p, TreeNode* q){
if(root==nullptr){
return false;
}
bool lson=dps(root->left,p,q);
bool rson=dps(root->right,p,q);
//该节点成为最近公共祖先要满足的条件;且该函数是向下递归遍历,所以得到的是最近祖先
if((lson && rson) || ((root->val == p->val || root->val == q->val) && (lson || rson))) {
res = root;
}
return lson || rson || (root->val == p->val || root->val == q->val);
}
*/
//思路二:用哈希表记录每个父节点,然后遍历判断
//用来存储每个结点的父节点的哈希表;一个是用来记录p、q结点的路径
unordered_map<int,TreeNode*> father;
unordered_map<int, bool> is;
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q){
father[root->val]=nullptr;
dfs(root);
while(p!=nullptr){
is[p->val]=true;
p=father[p->val];
}
//当有共同的祖父时,返回,否则继续往上遍历直至nullptr
while(q!=nullptr){
if(is[q->val]){
return q;
}
q=father[q->val];
}
return nullptr;
}
void dfs(TreeNode* root){
if (root->left != nullptr) {
father[root->left->val] = root;
dfs(root->left);
}
if (root->right != nullptr) {
father[root->right->val] = root;
dfs(root->right);
}
}
};