​LeetCode刷题实战236:二叉树的最近公共祖先

沙海
沙海
沙海
994
文章
2
评论
2021年4月15日01:12:37
评论
3 2567字阅读8分33秒
摘要

​LeetCode刷题实战236:二叉树的最近公共祖先 原创 小猿同学 程序IT圈

​LeetCode刷题实战236:二叉树的最近公共祖先

原创 小猿同学 程序IT圈

收录于话题

#Leecode刷题

110个

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从LeetCode上面选 !

今天和大家聊的问题叫做 二叉树的最近公共祖先,我们先来看题面:

https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/

Given a binary tree, find the lowest common ancestor (LCA) of two given nodes in the tree.

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

示例

​LeetCode刷题实战236:二叉树的最近公共祖先

解题

非递归算法

class Solution {public:    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {        stack<TreeNode *> stack;        stack.push(root);        map<TreeNode *, TreeNode *> parent_map;        parent_map[root] = NULL;        while(parent_map.find(p) == parent_map.end() || parent_map.find(q) == parent_map.end()){            TreeNode *node = stack.top();            stack.pop();            if(node -> left != NULL){                parent_map[node -> left] = node;                stack.push(node -> left);            }            if(node -> right != NULL){                parent_map[node -> right] = node;                stack.push(node -> right);            }        }        set<TreeNode *> p_parent;        p_parent.insert(p);// 迭代父节点map,这里的循环逻辑为,父节点字典中如果一直能够找到p节点的父节点,就进入循环// 这样可以一直得到p节点的父节点列表。        while(parent_map.find(p) != parent_map.end()){            p_parent.insert(parent_map

);            p = parent_map

;        }// 同理遍历q节点的父节点列表,因为是逆序遍历,所以当它存在于p节点的列表时,找到的就是p和q的最近公共祖先        while(parent_map.find(q) != parent_map.end()){            if(p_parent.find(q) != p_parent.end()){                return q;            }            q = parent_map[q];        }        return root;    }};

递归算法

class Solution {public:    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {        int num = contain_node(root, p, q);        return ans;    }    int contain_node(TreeNode* root, TreeNode* p, TreeNode* q){        if(!root) return 0;        int mid = 0;        if(root == p || root == q) mid = 1;        int left = contain_node(root -> left, p, q);        if(mid + left == 2){            if(!ans) ans = root;            return 2;        }        int right = contain_node(root -> right, p, q);        if(left + right + mid >= 2){            if(!ans) ans = root;            return 2;        }        return left + mid + right;    }private:    TreeNode* ans = NULL;};

好了,今天的文章就到这里,如果觉得有所收获,请顺手点个在看或者转发吧,你们的支持是我最大的动力 。

上期推文:

LeetCode1-220题汇总,希望对你有点帮助!

LeetCode刷题实战221:最大正方形

LeetCode刷题实战222:完全二叉树的节点个数

LeetCode刷题实战223:矩形面积

LeetCode刷题实战224:基本计算器

LeetCode刷题实战225:用队列实现栈

LeetCode刷题实战226:翻转二叉树

LeetCode刷题实战227:基本计算器 II

LeetCode刷题实战228:汇总区间

LeetCode刷题实战229:求众数 II

LeetCode刷题实战230:二叉搜索树中第K小的元素

LeetCode刷题实战231:2的幂

LeetCode刷题实战232:用栈实现队列

LeetCode刷题实战233:数字 1 的个数

LeetCode刷题实战234:回文链表

LeetCode刷题实战235:二叉搜索树的最近公共祖先

​LeetCode刷题实战236:二叉树的最近公共祖先

继续阅读
weinxin
资源分享QQ群
本站是一个IT技术分享社区, 会经常分享资源和教程; 分享的时代, 请别再沉默!
沙海
匿名

发表评论

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: