Java 圖文並茂講解兩種找二元樹的最近公共祖先的方法

public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    if(root == null) return null;

    // p 為當前樹的根節點
    if(p == root) return p;
    // q 為當前樹的根節點
    if(q == root) return q;
    
    // 去左子樹中找
    TreeNode left = lowestCommonAncestor(root.left,p,q);
    // 去右子樹中找
    TreeNode right = lowestCommonAncestor(root.right,p,q);

    // 左邊右邊都找到了
    if(left != null && right != null) {
        return root;
    }
    // 左邊找到了，右邊沒找到
    if(left != null) {
        return left;
    }
    if(right != null) {
        return right;
    }
    return null;
}

// 用於找節點的路徑
public boolean getPath(TreeNode root, TreeNode node, Stack<TreeNode> stack) {
    if(root == null || node == null) {
        return false;
    }
    // 將當前節點放入棧中
    stack.push(root);
    
    if(root.val == node.val) {
        return true;// 找到了
    }
    // 當前節點沒找到，去左子樹找
    boolean flag = getPath(root.left,node,stack);
    // 左子樹中找到了，直接返回
    if(flag) {
        return true;
    }
    // 左子樹沒找到，去右子樹找
    flag = getPath(root.right,node,stack);
    // 右子樹中找到了，直接返回
    if(flag) {
        return true;
    }
    
    // 左右子樹都沒找到，彈出節點
    stack.pop();
    return false;

}
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    if(root == null) {
        return null;
    }
    Stack<TreeNode> stackp = new Stack<>();
    Stack<TreeNode> stackq = new Stack<>();

    // 分別得到  p q 的路徑
    getPath(root,p,stackp);
    getPath(root,q,stackq);

    int sizep = stackp.size();
    int sizeq = stackq.size();

    if(sizep > sizeq) {
        int size = sizep - sizeq;
        // 彈出元素直至兩棧中元素個數相等
        while(size > 0) {
            stackp.pop();
            size--;
        }
    }else {
        int size = sizeq - sizep;
        // 彈出元素直至兩棧中元素個數相等
        while(size > 0) {
            stackq.pop();
            size--;
        }
    }

    // 一起彈出，直到找到第一個相同的元素
    while(!stackp.isEmpty() && !stackq.isEmpty()) {
        if(stackp.peek() == stackq.peek()) {
        	// 找到了，就返回該節點
            return stackq.pop();
        }else {
            stackp.pop();
            stackq.pop();
        }
    }
    // 沒找到，返回 null
    return null;
}