資料結構與算法:二叉搜尋樹的最近公共祖先

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) {
    }
}