天天看点

数据结构与算法:二叉搜索树的最近公共祖先

import java.util.ArrayList;
import java.util.List;

public class LowestCommonAncestorTree {

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

    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @param p int整型
     * @param q int整型
     * @return int整型
     */
    public int lowestCommonAncestor (TreeNode root, int p, int q) {
        List<Integer> pPath = getPath(root,p);  // 找到到达目标值p的路径
        List<Integer> qPath = getPath(root,q);  // 找到到达目标值q的路径

        int res = 0;
        for (int i = 0; i < pPath.size()&&i<qPath.size(); i++) {
            int x = pPath.get(i);
            int y = qPath.get(i);

            //最后一个相同的节点就是最近公共祖先
            if(x == y){
                res = pPath.get(i);
            }else{
                break;
            }
        }

        return res;
    }

    List<Integer> getPath(TreeNode root, int target){
        List<Integer> paths = new ArrayList<>();
        //节点值都不同,可以直接用值比较
        while(root.val != target){
            paths.add(root.val);
             if(root.val < target){
                 //大的在右子树
                root = root.right;
            }else if(root.val > target){
                 //小的在左子树
                root = root.left;
            }
        }

        paths.add(root.val);
        return paths;
    }

    public static void main(String[] args) {
    }
}