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.
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
示例
解题
非递归算法
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;};
好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力 。
上期推文：
继续阅读
资源分享QQ群
本站是一个IT技术分享社区, 会经常分享资源和教程; 分享的时代, 请别再沉默！
评论