题目描述:给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
示例 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 时,有三个查找渠道。
1)当前节点自身是不是p或q;
2)看左子树是否包含p或q;
3)看右子树是否包含p或q。 - 如果p和q来自于以上三个渠道中的两个,即可判断此节点是p和q的最近公共祖先。
代码实现:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
//创建成员变量记录最终结果
public TreeNode lca = null;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if(root == null){
return null;
}
//findNode 方法用来递归查找p和q
findNode(root,p,q);
return lca;
}
//若能找到p或q,返回true,都没找到,返回false
public boolean findNode(TreeNode root,TreeNode p,TreeNode q){
if(root == null){
return false;
}
//从以下三个渠道查找
//判断根节点是不是p或q
int mid = (root == p || root == q) ? 1 : 0;
//在左子树中查找
int left = findNode(root.left,p,q) ? 1 : 0;
//在右子树中查找
int right = findNode(root.right,p,q) ? 1 : 0;
//判断该节点是否是lca,即若p和q是来自于三个渠道中的两个即可。
if(mid + left + right == 2){
lca = root;
}
//方法返回值,找到p或q即为true
return (mid + left +right) > 0;
}
}