这道题想了挺久的,开始没思路,而且看leetcode的评论区几乎都是同一份短小精悍的代码,所以就打算自己先实现一个版本,再去看看分析评论区的代码,我的写法,如果使用多线程,应该比单线程快一些,可惜leetcode上这道题不支持使用多线程,使用多线程,锁住一下_set集合就可以了~
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
这道题我个人的解题思路是,从根节点root去遍历到每个叶子节点,如果找到p或者q,那么就把root到p或q的路径记录下来,则路径中最早出现两次的节点为其最近公共祖先。
在示例1的情况下,root到值为5的节点,路径列表里就有root和值为5的节点,root到值1的节点,路径列表就会加入值为1的节点与root节点,此时要注意的是,我们在加入节点前,都需要检测列表里是否存在该节点(或者计数该节点的出现次数),所以再次加入root节点时,列表里有该元素,那么root节点就为p与q的最近公共祖先。
在示例2的情况下,会出现,值为4的节点出现1一次,值为5的节点出现两次,值为3的节点出现两次,所以我们得取最先出现两次的节点为5,则p与q的最近公共祖先为5.
代码:
/**
* 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_set_set;
TreeNode *res;
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (p->val == root->val)
return p;
if (q->val == root->val)
return q;
res=NULL;
find(root,p);
find(root,q);
return res;
}
bool find(TreeNode* root,TreeNode *tar){
if(root==NULL)
return false;
if(root->val==tar->val){
if (_set.find(root) != _set.end())
res=root;
else
_set.insert(root);
return true;
}
bool f=false,k=false;
f=find(root->left,tar);
if(!f){
k=find(root->right,tar);
}
if(f||k){
if(_set.find(root)!=_set.end()){
if(!res)
res=root;
}
else
_set.insert(root);
}
return f||k;
}
};