题解 | #二叉搜索树的最近公共祖先#TOP37

思路:
二叉搜索树:左边的节点比根节点小,右边的节点比根节点大
1.递归。当两个节点分别在当前节点的左子树和右子树上时,该节点就是最近的公共祖先。题目要求是深度尽可能最大。否则我直接树顶部节点就是公共祖先,还做啥题目。如果当两个节点都在左边或者右边时,就继续递归求解即可
2.路径判断法。先求出两个节点的路径,直到找到这两个节点。然后遍历从根到节点的路径,如果值一样,那就是公共祖先了。因为值一样,节点一样,那路径就是这时候开始交叉的,按题目意思,同一层的节点,值相同必然是一个节点。(所有节点的值都不一样)

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
        if(root == null){
            return -1;
        }
        //return parent(root, p, q);
        return getCommonAncestor(root, p, q); 
    }
    private int parent(TreeNode root, int p, int q){
        if(root == null){
            return -1;
        }
        if((p >= root.val && q <= root.val) || (p<= root.val && q >= root.val)){
            return root.val;
        }else if(p <= root.val && q <= root.val){
            return lowestCommonAncestor(root.left, p, q);
        }else{
            return lowestCommonAncestor(root.right , p ,q);
        }
        
    }
    
    private int getCommonAncestor(TreeNode root, int p, int q){
        ArrayList<Integer> pathP = getPath(root, p);
        ArrayList<Integer> pathQ = getPath(root, q);
        int result = 0;
        for(int i = 0; i < pathP.size() && i <pathQ.size(); i++){
            int x = pathP.get(i);
            int y = pathQ.get(i);
            if(x == y){
                result = x;
            }else{
                break;
            }
        }
        return result;
    }
    
    private ArrayList<Integer> getPath(TreeNode root, int target){
        ArrayList<Integer> path = new ArrayList<>();
        TreeNode node = root;
        while(node.val != target){
            path.add(node.val);
            if(target < node.val){
                node = node.left;
            }else{
                node = node.right;
            }
        }
        path.add(node.val);
        return path;
    }
}
全部评论

相关推荐

孤寡孤寡的牛牛很热情:为什么我2本9硕投了很多,都是简历或者挂,难道那个恶心人的测评真的得认真做吗
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务