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

原创已于 2023-06-10 16:54:04 修改

· 1k 阅读

8 ·

版权

文章标签：

#算法 #数据结构 #leetcode

于 2023-03-26 20:46:00 首次发布

Java 面经专栏收录该内容

2 篇文章

订阅专栏

该文章介绍了一种从给定数组构建二叉树的方法，通过创建一个节点类，初始化树节点列表，然后根据数组中的值构建二叉树的结构。在遍历过程中，利用2i+1和2i+2的索引规则来设置左右子节点，确保不忽略任何有效节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

很多笔试都是白板（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;
	}
}