笔试—ACM模式构建二叉树输入输出

很多笔试都是白板（ACM模式），对于树的题目就需要自己构建二叉树

比如 题目给你一个数组，如何构建成二叉树的输入输出

class Solution{
	//节点类
	public class TreeNode {
	
		int val;
		TreeNode left;
		TreeNode right;
		
		TreeNode() {}
		
		TreeNode(int val) { 
			this.val = val; 
		}
		
		TreeNode(int val, TreeNode left, TreeNode right) {
			this.val = val;
			this.left = left;
			this.right = right;
		}
	}
	
	public TreeNode constructBinaryTree(final int[] arr){
		// 构建和原数组相同的树节点列表
		List<TreeNode> treeNodeList = arr.length > 0 ? new ArrayList<>(arr.length) : null;
		TreeNode root = null;
		// 把输入数值数组，先转化为二叉树节点列表
		for(int i  = 0; i < arr.length; i++){
			TreeNode node = null;
			if(arr[i] != -1){//用-1表示null
				node = new TreeNode(arr[i]);
			}
			treeNodeList.add(node);
			if(i == 0){
				root = node;
			}
		}
		//遍历一遍，根据规则左右孩子赋值就可以了
		//注意这里 结束规则是 i * 2 + 1 < arr.length，避免空指针
		//如果i * 2 + 2 < arr.length 是结束条件，那么i * 2 + 1这个符合条件的节点就被忽略掉了
		//例如[2,7,9,-1,1,9,6,-1,-1,10] 这样的一个二叉树,最后的10就会被忽略掉
		for(int i = 0; i * 2 + 1 < arr.length; i++){
			TreeNode node = treeNodeList.get(i);
			if(node != null){
				//线性存储转连式存储关键逻辑
				node.left = treeNodeList.get(2 * i + 1);
				//再次判断下 不忽略任何一个节点
				if(i * 2 + 2 < arr.length)
				node.right = treeNodeList.get(2 * i + 2);
			}
		}
		return root;
	}
}