【Leetcode】236. Lowest Common Ancestor of a Binary Tree

package medium;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import dataStructure.TreeNode;

public class L236LowestCommonAncestorOfaBT {
    
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // 求解方式：找到从根节点到两个叶子节点pq的路径，根据两条路径找到第一个公共子结点。
        if (root == null || root == p || root == q)
            return root;
        List<TreeNode> path1 = new ArrayList<TreeNode>();
        getPath(root, p, path1);
        List<TreeNode> path2 = new ArrayList<TreeNode>();
        getPath(root, q, path2);
        TreeNode ans = getCommanNode(path1, path2);
        return ans;
    }

    private TreeNode getCommanNode(List<TreeNode> path1, List<TreeNode> path2) {
        // TODO Auto-generated method stub
        int len1 = path1.size();
        int len2 = path2.size();
        int i;
        for (i = 0; i < len1 && i < len2; i++) {
            if (path1.get(i).val != path2.get(i).val) {
                return path1.get(i - 1);
            }
        }
        return path1.get(i - 1);
    }

    public boolean getPath(TreeNode root, TreeNode p, List<TreeNode> path) {
        // TODO 求根节点到p、q结点的路径
        path.add(root);
        if (root == p) {
            return true;
        }

        boolean found = false;
        if (root.left != null) {
            found = getPath(root.left, p, path);
            if (found == true)
                return true;
        }
        if (root.right != null) {
            found = getPath(root.right, p, path);
            if (found == true)
                return true;
        }

        path.remove(root);
        return found;
    }

    public static void main(String[] args) {
        TreeNode n1 = new TreeNode(1);
        TreeNode n2 = new TreeNode(2);
        TreeNode n3 = new TreeNode(3);
        n1.left = null;
        n1.right = n2;
        n2.left = null;
        n2.right = n3;

        TreeNode n4 = null;
        TreeNode n5 = null;
        n3.left = n4;
        n3.right = n5;
        L236LowestCommonAncestorOfaBT l236 = new L236LowestCommonAncestorOfaBT();
        //测试getPath()函数
        List<TreeNode> path1 = new ArrayList<TreeNode>();
        l236.getPath(n1, n3, path1);
        for (int i = 0; i < path1.size(); i++) {
            System.out.println(path1.get(i).val);
        }
        List<TreeNode> path2 = new ArrayList<TreeNode>();
        l236.getPath(n1, n2, path2);
        for (int i = 0; i < path2.size(); i++) {
            System.out.println(path2.get(i).val);
        }
        System.out.println("~~~~~~~~~~~~~~");
        //测试求公共结点函数 getCommanNode()
        TreeNode com = l236.getCommanNode(path1, path2);
        System.out.println("getCom祖先：" + com.val);
        System.out.println("~~~~~~~~~~~~~");
        //lowestCommonAncestor()函数
        TreeNode comTreeNode = l236.lowestCommonAncestor(n1, n2, n3);
        System.out.println("公共祖先为：" + comTreeNode.val);
        TreeNode node1 = new TreeNode(1);
        TreeNode node2 = new TreeNode(2);
        n1.left = null;
        n1.right = n2;
        n2.left = null;
        n2.right = null;
        System.out.println("~~~~~~~~~~~~~~~~~~");
        TreeNode comTreeNode1 = l236.lowestCommonAncestor(node1, node1, node2);
        System.out.println("case2：" + comTreeNode1.val);
    }
}

public TreeNode lowestCommonAncestor2(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q)
            return root;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);

        if (left != null && right != null)
            return root;
        return left != null ? left : right;
    }