Java数组转为二叉树-由LeetCode236题（求二叉树的最近公共祖先）引出的问题拓展

直接上题目
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

下图是该数组转换成二叉树的形状

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree

数组转二叉树方法Java代码，递归方法实现

public static TreeNode ArrToTree(Integer []arr, int index)
{
	//申请一个根结点
    TreeNode root = null;
    if (index<arr.length) {
        Integer value = arr[index];
        if (value == null) {
            return null;
        }
        //自顶向下构建二叉树的每一层
        root = new TreeNode(value);
        root.left = ArrToTree(arr, 2*index+1);
        root.right = ArrToTree(arr, 2*index+2);
        return root;
    }

    return root;
}

完成代码如下

package LeetCode236;

import java.util.*;

class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
}

public class Solution {
    static Map<Integer, TreeNode> parent = new HashMap<Integer, TreeNode>();
    static Set<Integer> visited = new HashSet<Integer>();

    public static void dfs(TreeNode root) {
        if (root.left != null) {
            parent.put(root.left.val, root);
            dfs(root.left);
        }

        if (root.right != null) {
            parent.put(root.right.val, root);
            dfs(root.right);
        }
    }

    public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        dfs(root);
        while (p != null) {
            visited.add(p.val);
            p = parent.get(p.val);
        }
        while (q != null) {
            if (visited.contains(q.val)) {
                return q;
            }
            q = parent.get(q.val);
        }
        return null;
    }

    public static TreeNode ArrToTree(Integer []arr, int index)
    {
        TreeNode root = null;
        if (index<arr.length) {
            Integer value = arr[index];
            if (value == null) {
                return null;
            }
            root = new TreeNode(value);
            root.left = ArrToTree(arr, 2*index+1);
            root.right = ArrToTree(arr, 2*index+2);
            return root;
        }
        return root;
    }

    public static void main(String[] args) {
        Integer nums[]={3,5,1,6,2,0,8,null,null,7,4};
        TreeNode root=ArrToTree(nums, 0);
        TreeNode p=new TreeNode(5);
        TreeNode q=new TreeNode(4);
        System.out.println(lowestCommonAncestor(root,p,q).val);
    }
}