牛客-TOP101-BM37

最新推荐文章于 2024-09-01 15:30:33 发布

一条吃猫的鱼

最新推荐文章于 2024-09-01 15:30:33 发布

阅读量90

点赞数

分类专栏：牛客-TOP101 文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/qq_43822715/article/details/125959526

版权

牛客-TOP101 专栏收录该内容

96 篇文章 0 订阅

订阅专栏

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param p int整型 
     * @param q int整型 
     * @return int整型
     */
    public int lowestCommonAncestor (TreeNode root, int p, int q) {
        // write code here
        ArrayList<Integer> pp = getPath(root, p);
        ArrayList<Integer> qq = getPath(root, q);
        int res = 0;
        for(int i = 0 ; i < pp.size() && i < qq.size(); i++){
            int x = pp.get(i);
            int y = qq.get(i);
            if(x == y)
                res = pp.get(i);
            else
                break;
        }
        return res;
    }
    
    public ArrayList<Integer> getPath(TreeNode root, int target) {
        ArrayList<Integer> list = new ArrayList<>();
        TreeNode node = root;
        while(node.val != target && node != null){
            list.add(node.val);
            if(node.val > target){
                node = node.left;
            }else{
                node = node.right;
            }
        }
        list.add(node.val);
        return list;
    }
}