代码随想录二刷 ｜ 二叉树 ｜ 二叉树的最近公共祖先

题目描述

236.二叉树的最近公共祖先

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

百度百科中最近公共祖先的定义为：“对于有根树 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 为不同节点且均存在于给定的二叉树中。

解题思路

遇到这个题目，首先想到是否可以自底向上查找，因此可以使用回溯。

后序遍历（左右中）就是天然的回溯过程，可以根据左右子树的返回值，来处理中节点的逻辑。

接下来就看如何判断一个节点是节点q和节点p的公共祖先。

情况一：

找到一个节点，发现左子树出现结点 p，右子树出现节点 q，或者 左子树出现结点 q，右子树出现节点 p，那么该节点就是节点 p 和 q 的最近公共祖先。

判断逻辑：

如果递归遍历遇到 q，就将 q 返回，遇到 p 就将 p返回，那么如果 左右子树的返回值都不为空，说明此时的中节点，一定是 q 和 p 的最近祖先。

情况二：

节点本身p(q)，它拥有一个子孙节点q§

判断逻辑：

如果递归遍历遇到 q，就将 q 返回，遇到 p 就将 p 返回，那么如果 左右子树的返回值都不为空，说明此时的中节点，一定是 q 和 p 的最近祖先。

递归三部曲：

1. 确定递归函数的参数和返回值
   如果我们需要让返回值告诉我们是否找到了p、q，因此返回值需要设置为bool
   然而我们还要返回最近公共节点，因此可以利用题目给的TreeNode*，如果遇到 p 或者 q，就把 q 或者 p 返回，返回值不为空，就说明找到了q或者p。

   TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q)
   

2. 确定递归函数的终止条件
   遇到遍历时遇到空的话，递归函数就要终止了。

   因为树都是空了，所以返回空。

   同时，因为我们要寻找p、q，所以当root = q或root = p时，要将其返回，所以终止条件为：

   if (root == q || root == p || root == NULL) return root;
   

3. 确定单词递归的逻辑

   本题函数有返回值，是因为回溯的过程需要递归函数的返回值做判断，但本题我们依然要遍历树的所有节点。

   在递归函数有返回值的情况下：

   • 如果要搜索一条边，在递归函数返回值不为空时，立刻返回；

     if (递归函数(root->left)) return;
     
     if (递归函数(root->right)) return;
     

   • 如果搜索整个树，用变量left、right接住返回值，这个left、right在后面还有逻辑处理的需要，即后序遍历中处理中间节点的逻辑（也是回溯）。

     left = 递归函数(root->left); // 	左
     right = 递归函数(root->right); // 中
     left与right的逻辑处理 // 右
     

   直觉来说，找到公共组先后，遍历一条边回溯就可以了，但实际上需要遍历整棵树，因为我们需要递归函数的返回值来进行逻辑处理部分。

代码实现

class Solution {
public:
	TreeNode* lowerCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
		if (root == q || root == q || root == NULL) return root;
		TreeNode* left = lowerCommonAncestor(root->left, p, q);
		TreeNode* right = lowerCommonAncestor(root->right, p, q);
		// 此时root就是最近公共组线
		if (left != NULL && right != NULL) return root;
		// 情况二，此时right为最近公共组线
		if (left == NULL && right != NULL) return right;
		// 情况二，此时left为最近公共组线
		else if (left != NULL && right == NULL) return left;
		else return NULL; // left == NULL && right == NULL
	}
};