力扣算法31——JZ86 在二叉树中找到两个节点的最近公共祖先

最新推荐文章于 2024-10-17 19:06:39 发布

不想写代码的程序猿.

最新推荐文章于 2024-10-17 19:06:39 发布

阅读量192

点赞数

文章标签： leetcode 算法散列表

本文链接：https://blog.csdn.net/Sdwq12/article/details/125052504

版权

本文通过代码实现解析力扣题目JZ86，在二叉树中寻找两个节点的最近公共祖先。利用了Map的键值对特性和Set的不重复特性，结合队列进行层次遍历来解决此问题。

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

代码实现：

利用了map集合的kv特性
利用了set的不可重复特性
使用队列进行层次遍历

import java.util.*;
public class Solution {
    public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
        //创建一个map存放本节点和其父节点
        Map<Integer,Integer> parent = new HashMap<>();
        //创建一队列进行层次遍历
        Queue<TreeNode> queue = new LinkedList<>();
        //根节点没有父节点用nmin代替
        parent.put(root.val,Integer.MIN_VALUE);
        //放入队列
        queue.add(root);
        //当map中有o1和o2的key则跳出循环
        while(!parent.containsKey(o1) || !parent.containsKey(o2)){
            TreeNode p = queue.poll();
            if(p.left != null){
                parent.put(p.left.val,p.val);
                queue.add(p.left);
            }
            if(p.right != null){
                parent.put(p.right.val,p.val);
                queue.add(p.right);
            }
        }
        //创建一个set放父节点
        Set<Integer> set = new HashSet<>();
        //把o1和其父节点放入set
        while(parent.containsKey(o1)){
            set.add(o1);
            //拿的是o1的祖节点
            o1 = parent.get(o1);
        }
        //判断set是否包含o2，不相等就看o2的父节点
        while(!set.contains(o2))
            //取o2的父节点
            o2 = parent.get(o2);
        return o2;
    }
}