NC102 在二叉树中找到两个节点的最近公共祖先 Java算法

描述
给定一棵二叉树(保证非空)以及这棵树上的两个节点对应的val值 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点

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 helper(root, o1, o2).val;
    }
    
    public TreeNode helper(TreeNode root, int o1, int o2){
        if(root == null || root.val == o1 || root.val == o2){
            return root;
        }
        TreeNode left = helper(root.left, o1, o2);
        TreeNode right = helper(root.right, o1, o2);
        // 如果left为空,说明这两个节点在root节点右子树上,只需要返回右子树查询的结果即可
        if(left == null){
            return right;
        }
        // 如果right为空,说明这两个节点在root节点左子树上,只需要返回左子树查询的结果即可
        if(right == null){
            return left;
        }
        // 如果右子树和左子树都不为空,说明这两个节点一个在root的左子树上,一个在root的右子树上
        // 只需要返回当前节点即可
        return root;
    }
}
上一篇:list的排序


下一篇:CSP之检测点查询