文章目录
- 题目描述
- 思路
- 扩展
题目描述
给定一颗二叉树，以及2个指定节点p , q，找到这2个节点的最近公共祖先。注意，一个节点也可以是它自己的祖先。并且根据提示，有p != q，并且p和q均存在于二叉树中。
思路
由于p和q一定存在于树中，则其最近公共祖先一定存在。其最近公共祖先有如下2种情况
- 最近公共祖先是
p或者q - 最近公共祖先不是
p或者q
对于情况1，假设最近公共祖先是p，则q一定存在于p的左子树或右子树，此时直接返回p即可。当最近公共祖先是q时同理。
对于情况2，假设最近公共祖先是r，则p和q一定分别位于r的左子树和右子树上。
由于求解的是这种节点的父子关系，很容易想到需要用DFS+回溯的方式。我们期望在DFS的过程中，能够把节点p和q的信息，通过回溯传递上来。
所以我们这样来定义递归函数：
用dfs(node) 这个函数表示，对以node为根节点的二叉树进行搜索，若p在树中出现，则返回p；若q在树中出现，则返回q；若p和q同时在树中出现，则返回p和q的最近公共祖先。
伪代码为：
dfs(node):
if (node == null) return null; // 跨过叶子节点
if (node == p || node == q) return node; // 找到p或者q
left = dfs(node.left); // 查找node的左子树, 看p或q是否出现
right = dfs(node.right); // 查找node的右子树, 看p或q是否出现
if (left != null && right != null) return node; // 当 p 和 q 分别在 node 的左子树和右子树出现, 则最近公共祖先就是 node
if (left != null) return left; // 若左子树上找到了p或q, 返回左子树的查找结果
return right; // 右子树上找到了p或q，返回右子树的查找结果假设最近公共祖先不是p或q，是r。并且假设p位于r左子树，q位于r右子树。则通过对r.left进行dfs搜索，能够把p的节点信息传递上来，对r.right进行dfs搜索，能够把q的节点信息传递上来。并且r节点，是唯一一个对其左子树进行dfs有结果，并且对其右子树进行dfs也有结果的节点。（r节点下面的节点，最多只可能在某一侧的dfs中有结果；r节点上面的节点，只会在r所在的这一侧的子树中，有dfs的结果）。
假设最近公共祖先是p或q，则同样在遇到p或q时，dfs函数就提前返回了，和本题的答案相一致。
注意该方法中，对递归函数dfs的定义，其实是带有扩展的，它求解的是：
- 若
p出现在二叉树中，则返回p - 若
q出现，则返回q - 若
p和q同时出现，则返回二者的最近公共祖先 - 若
p和q都不出现，返回null
而由于本题中p和q一定存在于二叉树中，所以该方法能求出正确结果。
dfs函数在整个搜索过程中的作用是，将p或q的信息，通过返回值，传递上来（从子节点，往上传递给父节点，回溯）。
代码如下：
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null || root == p || root == q) return root;
TreeNode left = lowestCommonAncestor(root.left, p, q);
TreeNode right = lowestCommonAncestor(root.right, p, q);
if (left != null && right != null) return root;
return left != null ? left : right;
}
}扩展
假设p和q不一定都存在于二叉树当中，那么上面的方法就无法求出正确结果了。
我们换一种思路，我们设一个外部变量ans，来保存在dfs过程中可能搜到的答案，并且这样定义dfs函数：
dfs(node)表示：在以node为根节点的二叉树中进行搜索，若树中出现了p或者q，则返回true，否则返回false。
我们从根节点root开始进行DFS：
- 当
root等于p或q时，只需要递归的搜索root的左子树和右子树，只要其中一侧的dfs返回了true，则说明找到了另一个节点，此时更新ans = root，然后提前返回即可（最近公共祖先是p或者q的情况） - 当
root不等于p或q时，递归搜索两侧，只有当两侧的dfs都返回了true时，更新ans = root（最近公共祖先不是p或者q的情况，此时的root节点，是唯一一个对其左右子树调用dfs都返回true的节点） - 当
ans != null 时，说明已经找到答案，直接提前返回
另外要注意的是，假设root = p，且对root的子树调用dfs返回true，则说明在root子树中找到的一定是节点q；类似地，假设root不为p也不为q，若对root的左子树和右子树分别调用dfs都返回true，则一定是p和q分别位于root的左右子树当中。
也就是说，当找到了一个节点时，对另外一边调用dfs如果返回true，则一定是找到了另外一个节点。（同一个节点不可能存在于一棵树上的两个位置）
这种解法能够在p或 q不存在于二叉树的情况下，返回正确结果（null）。
class Solution {
TreeNode ans = null;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
dfs(root, p, q);
return ans;
}
private boolean dfs(TreeNode cur, TreeNode p, TreeNode q) {
if (cur == null || ans != null) return false; // 跨过叶子节点, 或者已经找到答案, 直接返回
if (cur == p || cur == q) {
if (dfs(cur.left, p, q) || dfs(cur.right, p, q)) {
ans = cur;
}
return true;
}
boolean left = dfs(cur.left, p, q);
boolean right = dfs(cur.right, p, q);
if (left && right) {
ans = cur;
}
return left || right;
}
}
