leetcode刷题(41)——236. 二叉树的最近公共祖先
一、题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4] 说明:
所有节点的值都是唯一的。 p、q 为不同节点且均存在于给定的二叉树中。
二、思路
思路就是不断地去找 p 和 q 的位置,到底在哪个子树中,找的过程中,可能会返回null、p、q、普通 Node。
-
返回 p、q 就说明 p、q 在当前节点的子树中(或当前节点就是); 返回普通 Node 说明找到了 p、q 的最近公共祖先; 返回 null 说明 p、q 不在该节点的子树。
寻找的过程中,首先是判断当前 root。如果满足 root 为 null 或为 p 或为 q ,则直接返回。之后会递归判断 root 的 left,进行剪枝判断是否需要递归 root.right,如果需要则会递归判断 root.right。
剪枝函数(递归结束条件):
if(root==null||root==p||root == q) return root;
如果 root 为 p 或 q 就可以直接返回,不需要判断 root 的左右子树,因为即使另一个节点在 root 的左子树,root 也是他们的最近公共节点。
三、代码
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) return right; else if(right == null) return left; else return root; } }