leetcode刷题(41)——236. 二叉树的最近公共祖先

一、题目

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
leetcode刷题(41)——236. 二叉树的最近公共祖先_第1张图片
说明:

所有节点的值都是唯一的。
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;
    }
}

你可能感兴趣的:(leetcode刷题,剪枝,二叉树,leetcode)