题目链接:236. 二叉树的最近公共祖先
题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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
均存在于给定的二叉树中。
文章讲解:代码随想录
视频讲解:自底向上查找,有点难度! | LeetCode:236. 二叉树的最近公共祖先_哔哩哔哩_bilibili
题解1:递归法
思路:后序遍历,自底向上返回当前节点下面是否有 p 和 q,最后得出结果。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @param {TreeNode} p
* @param {TreeNode} q
* @return {TreeNode}
*/
var lowestCommonAncestor = function(root, p, q) {
if (!root) {
return null;
}
if (root === p || root === q) {
return root; // 如果当前节点是 p 或 q,直接返回当前节点
}
const left = lowestCommonAncestor(root.left, p, q);
const right = lowestCommonAncestor(root.right, p, q);
if (left && right) {
return root; // 如果当前节点下面有 p 和 q,直接返回当前节点
}
if (left) {
return left; // 只有 p,则将有 p 的节点返回上去
}
if (right) {
return right; // 只有 q,则将有 q 的节点返回上去
}
return null; // 下面没 p 没 q,返回空
};
分析:时间复杂度为 O(n),空间复杂度为 O(logn)。
题解2:哈希表
思路:使用哈希表存储二叉树中每个节点的父节点,从 p 节点一直往上,访问直到根节点的这条路径上的节点,再从 q 节点一直往上,检查直到根节点的这条路径上的节点是否被访问过,第1个被访问过的节点即为 p 和 q 的最近公共祖先。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @param {TreeNode} p
* @param {TreeNode} q
* @return {TreeNode}
*/
var lowestCommonAncestor = function(root, p, q) {
const parent = {}; // 存储每个节点的父节点
const visit = {}; // 存储每个节点的访问状态
const preorder = function (node) {
if (node.left) {
parent[node.left.val] = node;
preorder(node.left);
}
if (node.right) {
parent[node.right.val] = node;
preorder(node.right);
}
};
parent[root.val] = null;
preorder(root); // 构建父节点哈希表
while (p) {
visit[p.val] = true; // 从 p 节点一直往上,访问直到根节点的这条路径上的节点
p = parent[p.val];
}
while (q) {
if (visit[q.val]) {
return q; // 从 q 节点一直往上,检查直到根节点的这条路径上的节点是否被访问过,第1个被访问过的节点即为 p 和 q 的最近公共祖先
}
q = parent[q.val];
}
return root;
};
分析:时间复杂度为 O(n),空间复杂度为 O(n)。
收获
自顶向下遍历二叉树使用前序遍历,自底向上遍历二叉树使用后序遍历。