【LeetCode】二叉树的最近公共祖先 [M]（二叉树递归）

最新推荐文章于 2024-07-22 16:27:42 发布

小七mod

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量280

点赞数

分类专栏： # LeetCode # 数据结构 # 算法文章标签： leetcode 算法数据结构递归 Java

本文链接：https://blog.csdn.net/cy973071263/article/details/128659652

版权

LeetCode 同时被 3 个专栏收录

192 篇文章 6 订阅

订阅专栏

算法

166 篇文章 0 订阅

订阅专栏

数据结构

78 篇文章 37 订阅

订阅专栏

236. 二叉树的最近公共祖先 - 力扣（LeetCode）

一、题目

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

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

示例 1：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出：3
解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出：5
解释：节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

示例 3：
输入：root = [1,2], p = 1, q = 2
输出：1

提示：

树中节点数目在范围 [2, 10^5] 内。
-10^9 <= Node.val <= 10^9
所有 Node.val 互不相同。
p != q
p 和 q 均存在于给定的二叉树中。

二、代码

/**
 * 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) {
         return process(root, p, q).ans;
    }

    // 信息类
    public static class Info {
        // 当前树有没有发现a节点
        public boolean findA;
        // 当前树有没有发现b节点
        public boolean findB;
        // 当前树找到的a和b的最低公共祖先是谁，如果没有在这棵树找到则设置为null
        public TreeNode ans;
        public Info(boolean fA, boolean fB, TreeNode an) {
            findA = fA;
            findB = fB;
            ans = an;
        }
    }

    // 二叉树递归套路
    public static Info process(TreeNode x, TreeNode a, TreeNode b) {
        // 1、处理空节点
        // 如果为空节点，则说明该节点没有a和b，也没有找到二者的最低公共祖先，返回相应的info
        if (x == null) {
            return new Info(false, false, null);
        }
        // 2、通过二叉树递归去获取左右子树的info
        Info leftInfo = process(x.left, a, b);
        Info rightInfo = process(x.right, a, b);
        // 3、通过左右子树的info来计算当前树的info所需要的信息
        // 根据所有的可能性求解
        // 如果当前x就是a,或者左子树发现了a或者右子树发现了a，就说明当前树发现了a
        boolean findA = (x == a) || leftInfo.findA || rightInfo.findA;
        // 同理
        boolean findB = (x == b) || leftInfo.findB || rightInfo.findB;
        // 先将ans设置为null
        TreeNode ans = null;
        // 如果左子树已经找到答案了，则当前树的答案就是左子树的答案
        if (leftInfo.ans != null) {
            ans = leftInfo.ans;
        // 如果右子树已经找到答案了，则当前树的答案就是右子树的答案 
        } else if (rightInfo.ans != null) {
            ans = rightInfo.ans;
        // 除了上面两种情况，然后发现a和b节点都在x树上，那么说明a和b的最低公共祖先一定是x   
        } else {
            // 如果上面的leftInfo.ans和rightInfo.ans都是空，说明一定不存在a和b都在左子树或者都在右子树的情况
            // 那么如果在X这棵树上还发现了a和b，那么只有一种可能，就是a和b分别在X的左右子树上
            // 这种情况就说明a和b的最低公共祖先一定是x
            if (findA && findB) {
                ans = x;
            }
        }
        // 4、返回当前树的info
        return new Info(findA, findB, ans);
    }

}