Java教程

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

本文主要是介绍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;
    }
}
这篇关于NC102 在二叉树中找到两个节点的最近公共祖先 Java算法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!