牛客网：NC102 在二叉树中找到两个节点的最近公共祖先

最新推荐文章于 2024-09-28 18:08:30 发布

what is an object

最新推荐文章于 2024-09-28 18:08:30 发布

阅读量112

点赞数

分类专栏：菜鸟刷题记录算法文章标签：算法 java 数据结构

本文链接：https://blog.csdn.net/newOneObject/article/details/125719149

版权

算法同时被 2 个专栏收录

278 篇文章 1 订阅

订阅专栏

菜鸟刷题记录

204 篇文章 1 订阅

订阅专栏

递归写法：

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    // 递归解法, 
    public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
        // write code here
        return help(root, o1, o2).val;
    }
    
    private TreeNode help(TreeNode root, int x, int y){
        if (root == null || root.val == x || root.val == y){
            return root;
        }
        TreeNode left = help(root.left, x, y);
        TreeNode right = help(root.right, x, y);
        if (left == null){
            return right;
        }
        if (right == null){
            return left;
        }
        return root;
    }
}

迭代写法：

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    // 记录路径
    public int lowestCommonAncestor (TreeNode root, int m, int n) {
        // write code here
        Map<Integer ,Integer> map = new HashMap<>();
        Queue<TreeNode> deque = new LinkedList<>();
        map.put(root.val, Integer.MAX_VALUE);
        deque.add(root);
        while (!map.containsKey(m) || !map.containsKey(n)){
            TreeNode temp = deque.poll();
            if (temp.left != null){
                deque.add(temp.left);
                map.put(temp.left.val, temp.val);
            }
            if (temp.right != null){
                deque.add(temp.right);
                map.put(temp.right.val, temp.val);
            }
        }
        Set<Integer> set = new HashSet<>();
        // 找到节点1的所有父节点
        while (map.containsKey(m)){
            set.add(m);
            m = map.get(m);
        }
        
        while(!set.contains(n)){
            n = map.get(n);
        }
        return n;
        
     }
}