给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 为不同节点且均存在于给定的二叉树中。
CODE:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private TreeNode ans = null;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
this.dfs(root, p, q);
return this.ans;
}
public boolean dfs(TreeNode node, TreeNode p, TreeNode q) {
if (node == null) {
return false;
}
boolean lson = this.dfs(node.left, p, q);
boolean rson = this.dfs(node.right, p, q);
if ((lson && rson) || ((node.val == p.val || node.val == q.val) && (lson || rson))) {
this.ans = node;
}
return lson || rson || node.val == p.val || node.val == q.val;
}
}
题解:
主要思考满足条件的情况:
一是两个点不在同一分支情况下,最近公共祖先在哪,就是最近的叉路口,条件就是lson&&rson
二是两个点如果在同一侧情况,最近公共祖先在哪,其实就是有一个是另一个的祖先,条件就是(p.val==val || q.val == val)&& (lson || rson)