题目
给定一棵二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义: “对于有根树T的两个结点u、v,最近公共祖先表示一个结点x,满足x是u、v的祖先且x的深度尽可能大。”(一个节点也可以是它自己的祖先)
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
_______3______ / \ ___5__ ___1__ / \ / \ 6 _2 0 8 / \ 7 4
示例 1:
输入: root, p = 5, q = 1 输出: 3 解释: 节点5
和节点1
的最近公共祖先是节点3。
示例 2:
输入: root, p = 5, q = 4 输出: 5 解释: 节点5
和节点4
的最近公共祖先是节点5。
因为根据定义最近公共祖先节点可以为指定节点自身。
分析
一开始想着从结点向上遍历父节点,,,后来忽然发现这个不可行。
有三种情况,p q 都在左子树或右子树,p q分别在左子树和右子树,
判断当前遍历的节点是否为空,为空返回null,节点是否等于p,是否等于q,是的话返回null。
之后判断left 和right是否为空, 若都不为空 ,则当前root为答案,若其中一个为空,则返回另一边,为答案。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
if (root == null) return null;
if (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;//p q 分别位于左右子树的情况
return left == null ? right : left;
}
}
有一点点不太明白的是,函数传入的TreeNode p,和TreeNode q,是什么样子的,
当
if (root == p || root == q) return root;
没有错误,
而如果改成
if (root.val == p.val || root.val == q.val) return root;
则会有通不过的数据。