LeetCode:二叉树的最近公共祖先

  这道题想了挺久的,开始没思路,而且看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<TreeNode*>_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;
    }
};

在这里插入图片描述

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页