【无标题】

最新推荐文章于 2024-08-14 22:49:08 发布

Time-leaper

最新推荐文章于 2024-08-14 22:49:08 发布

阅读量100

点赞数

文章标签：深度优先算法

本文链接：https://blog.csdn.net/cainiao00078/article/details/130526054

版权

文章提供了两种方法来寻找二叉树中两个给定节点的最低公共祖先。方法1使用深度优先搜索(DFS)，方法2通过迭代并保存每个节点的父节点。在DFS方法中，递归地检查节点及其子节点，当找到p和q时返回最近公共祖先。而在迭代方法中，先遍历整个树构建父节点映射，然后从p和q开始回溯至公共祖先。

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

方法1：dfs

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    TreeNode res;
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        boolean flag = help(root, p, q);
        return res;
    }

    public boolean help(TreeNode root, TreeNode p, TreeNode q){
        if(root == null){
            return false;
        }
        boolean left = help(root.left, p, q);
        boolean right = help(root.right, p ,q);
        if(left && right || (root==p && (left || right)) || (root==q &&(left || right))){ //如果当前节点的左子树和右子树分别存在p和q，p和q之一为当前节点，则当前节点为其最近祖先
            res = root;
            return true;
        }
        return root==p || root==q || left || right;
    }
}

方法2：迭代保存父节点

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //迭代保存parent节点
    Map<TreeNode, TreeNode> parents = new HashMap<>();
    Set<TreeNode> visit = new HashSet<>();
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        dfs(root);
        while(p!=null){
            visit.add(p);
            p = parents.get(p);
        }
        while(q!=null){
            if(visit.contains(q)){
                return q;
            }
            q = parents.get(q);
        }
        return null;
    }

    public void dfs(TreeNode root){
        if(root == null){
            return;
        }
        if(root.left != null){
            parents.put(root.left, root);
            dfs(root.left);
        }
        if(root.right != null){
            parents.put(root.right, root);
            dfs(root.right);
        }
    }
}