【dfs】二叉树的最近公共祖先

最新推荐文章于 2024-10-05 22:19:52 发布

冬瓜的编程笔记

最新推荐文章于 2024-10-05 22:19:52 发布

阅读量470

点赞数 10

分类专栏：刷题笔记文章标签：深度优先算法

本文链接：https://blog.csdn.net/m0_62332728/article/details/136077208

版权

刷题笔记专栏收录该内容

30 篇文章 0 订阅

订阅专栏

文章介绍了如何通过两次遍历二叉树的方法，在先序遍历过程中查找最近公共祖先。作者使用`ans`变量存储公共祖先，同时利用`pTrue`和`qTrue`标记节点是否为p或q，确保找到最近的公共祖先。

摘要由CSDN通过智能技术生成

题目：

思路：

两次遍历二叉树，首先遍历全部的二叉树节点。在遍历的同时判断每个二叉树的节点是否为p,q的公共祖先。如果是公共祖先用一个变量ans 存储该节点。然后重复此过程最终ans一定是最近公共最先。 为什么ans最终一定为最近公共祖先：我想的是在遍历的过程中一直在找全部的公共最先并且第一次遍历是先序遍历，是从根节点逐渐向下遍历。所以找到的公共祖先离p,q越来越近。（至于怎么证明我不会）


/**
 * @auther start
 * @create 2024-02-08 10:32
 */
public class L236 {
    //ans存储公共最先
    TreeNode ans;
    //如果找到了p,q，使用这两个变量标记
    boolean pTrue, qTrue;
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        ans = new TreeNode();
        dfsTmp(root,p,q);
        return ans;
    }

    //判断节点树中是否存在p,q
    public void dfs(TreeNode root, TreeNode p, TreeNode q){
        if (root == null) return;
        if (root == p) pTrue = true;
        if (root == q) qTrue = true;
        dfs(root.left,p,q);
        dfs(root.right,p,q);

    }
    //第一次遍历，遍历全部的节点，先序遍历
    public void dfsTmp(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null) return;;
        //判断这个节点树是否有p,q
        dfs(root,p,q);
        //如果这个节点树中有p和q使用ans保存
        if (pTrue && qTrue) ans = root;
        //恢复变量为默认状态
        pTrue = false;
        qTrue = false;
        dfsTmp(root.left,p,q);
        dfsTmp(root.right,p,q);
    }
}