题目
给定一棵二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义: “对于有根树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;则会有通不过的数据。