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 。
思路:用后序遍历的方法,自底向上去遍历二叉树,遇到p或q节点就返回p或q。
递归法:
- 确定递归函数参数列表和返回值:参数列表为根节点,p节点和q节点,返回值为要返回的最近公共祖先,类型为TreeNode*。
- 确定终止条件:如果当前节点是p或q或者空节点,就返回当前节点。
- 确定单层递归逻辑:分别在左子树和右子树中找公共祖先,如果左子树与右子树都不为空,说明p和q分布在两个子树中,此时当前节点就是最近公共祖先,返回当前节点;如果左子树为空,说明p和q都在右子树中,返回右子树的返回值;否则,返回左子树的返回值。
代码:
class Solution { //236. 二叉树的最近公共祖先
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == p || root == q || root == nullptr) return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (left != nullptr && right != nullptr)return root;
if (left == nullptr)return right;
return left;
}
};
int main() {
TreeNode* p1 = new TreeNode(37);
TreeNode* p2 = new TreeNode(-34);
TreeNode* p3 = new TreeNode(-48);
TreeNode* p4 = new TreeNode(-100);
TreeNode* p5 = new TreeNode(-101);
TreeNode* p6 = new TreeNode(48);
TreeNode* p7 = new TreeNode(-54);
TreeNode* p8 = new TreeNode(-71);
TreeNode* p9 = new TreeNode(-22);
TreeNode* p10 = new TreeNode(8);
p1->left = p2;
p1->right = p3;
p2->right = p4;
p3->left = p5;
p3->right = p6;
p6->left = p7;
p7->left = p8;
p7->right = p9;
p9->right = p10;
Solution s;
TreeNode* result = s.lowestCommonAncestor(p1, p8, p10);
return 0;
}
参考资料:代码随想录