【每日一题】【DFS】2022年1月14日-NC102 在二叉树中找到两个节点的最近公共祖先

描述

给定一棵二叉树(保证非空)以及这棵树上的两个节点对应的val值 o1 和 o2，请找到 o1 和 o2 的最近公共祖先节点。 数据范围：1 \le n \le 10001≤n≤1000，树上每个节点的val满足 00<val≤100 要求：时间复杂度 O(n)O(n) 注：本题保证二叉树中每个节点的val值均不相同。 如当输入[3,5,1,6,2,0,8,#,#,7,4],5,1时，二叉树{3,5,1,6,2,0,8,#,#,7,4}如下图所示：

所以节点值为5和节点值为1的节点的最近公共祖先节点的节点值为3，所以对应的输出为3，节点本身可以视为自己的祖先

方法1：DFS-找左找右

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) {
        return dfs(root, o1, o2).val;
    }
    
    public TreeNode dfs(TreeNode root, int o1, int o2) {
        if(root == null) {
            return null;
        }
        if(root.val == o1 || root.val == o2) {
            return root;
        }
        TreeNode left = dfs(root.left, o1, o2);
        TreeNode right = dfs(root.right, o1, o2);
        if(left != null && right != null) {
            return root;
        } 
        return left == null ? right : left;
    }
}

所以节点值为5和节点值为1的节点的最近公共祖先节点的节点值为3，所以对应的输出为3。节点本身可以视为自己的祖先