Leetcode 236 Lowest Common Ancestor of a Binary Tree(最近的公共祖先)

Lowest Common Ancestor of a Binary Tree
Example 1:
Input: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
Output: 3
Input: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
Output: 5

在这里插入图片描述

出口:在BST中的思路是如果p和q一个大于节点值,一个小于节点值,那么该节点一定就是最近的
祖先节点
若不是则(小于)去左子树(大于)右子树中找,直到找到上述出口所述状态
如果没有BST这个前提,那么就不能用pq分别大于小于作为出口,那么这样的出口在哪呢
出口:节点为空,节点为p,或q
1. 左右子树的返回值都不为null, 由于值唯一左右子树的返回值就是p和q, 此时root为LCA
2. 如果左右子树返回值只有一个不为null, 说明只有p和q存在与左或右子树中, 最先找到的那个节点为LCA
3. 左右子树返回值均为null, p和q均不在树中, 返回null

思路
所有的递归的返回值有4种可能性,null、p、q、公共祖先
两个节点p,q分为两种情况:

p和q在相同子树中
p和q在不同子树中

从根节点遍历,递归向左右子树查询节点信息
递归终止条件:如果当前节点为空或等于p或q,则返回当前节点

递归遍历左右子树,如果左右子树查到节点都不为空,
则表明p和q分别在左右子树中,因此,当前节点即为最近公共祖先
如果左右子树其中一个不为空,则返回非空节点。
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root==null||root==p||root==q){
            return root;
        }//首先对root进行特殊考虑
        TreeNode left=lowestCommonAncestor(root.left,p,q);
        TreeNode right=lowestCommonAncestor(root.right,p,q);
        
        //root是公共节点的情况(此时left right分别是p、q)
        if(left!=null&&right!=null){
            return root;
        }
        if(left!=null){
            return left;
        }
        if(right!=null){
            return right;
        }
        return null;
    }
}
 
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (!root || root == p || root == q) return root;
        TreeNode *left = lowestCommonAncestor(root->left, p, q);
        TreeNode *right = lowestCommonAncestor(root->right, p, q);
        if (left && right) return root;//分在左右子树的状况
        return left ? left : right;//分在同一颗子树的情况
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值