题解 | #二叉搜索树的最近公共祖先#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;
}
}