ACM 构造二叉树输入用例
在今年的秋招面试中,面试官要构造二叉树,输入用例就给了一个数组,怎么就能构造成二叉树呢?
(可以先跳过,如果出现了类似的情况再来看)之后自己java写了一个完整的可执行程序。书写的时候出了两处bug:
-
构造二叉树的函数只返回了root节点,而不是整个二叉树;
-
原因:
1.root和二叉树失去了联系:root=new TreeNode();
2.数组不能为偶数,因为数组表示二叉树默认这棵二叉树是完美的
完整代码如下:
package?代码随想率;
import?java.util.Arrays;
import?java.util.LinkedList;
import?java.util.Queue;
public?class?构建二叉树?{
????public?static?void?main(String[]?args)?{
????????int?a[]={1,2,3,4,5,6};//数组不能为偶数,因为数组表示二叉树默认这棵二叉树是完美的
????????TreeNode?root?=?construstBinaryTree(a);
????????System.out.println("层次遍历之后结果:");//?I?get??an?Error:nullpointException?!
????????levelRead(root);
????}
????//构造二叉树并返回root节点
????public?static??TreeNode?construstBinaryTree(int?[]?nums){
????????TreeNode?treeNodeArray[]=new?TreeNode[nums.length];
????????Arrays.fill(treeNodeArray,null);
????????TreeNode?root?=?null;
????????for?(int?i?=?0;?i?<?nums.length;i++)?{
??????????if(nums[i]?!=?-1)?{treeNodeArray[i]?=?new?TreeNode(nums[i]);}
??????????if(i?==?0)?{
??????????????root?=?treeNodeArray[0];//注意这里是treeNodeArray[0],而不是new?TreeNode(nums[0])
??????????}
????????}
????????System.out.print("二叉结点数组:");
????????for?(TreeNode?treeNode?:?treeNodeArray)?{
????????????System.out.print(treeNode.val+"?");
????????}
????????System.out.println();
????????//二叉树数组->二叉树
????????for?(int?i?=?0;?2?*?i?+?2?<?nums.length;?i++){
????????????if(treeNodeArray[i]?!=?null)?{
????????????????treeNodeArray[i].left?=?treeNodeArray[2?*?i?+?1];
????????????????treeNodeArray[i].right?=?treeNodeArray[2?*?i?+?2];
????????????}
????????}
????????return?root;
????}
????//构造二叉树并返回root节点
????public?static?void?levelRead(TreeNode?root)
????{
????????if(root?==?null)?return;
????????Queue<TreeNode>?queue?=?new?LinkedList<TreeNode>()?;
????????queue.add(root);
????????while(queue.size()?!=?0)
????????{
????????????int?len?=?queue.size();
????????????for(int?i=0;i?<len;?i++)
????????????{
????????????????TreeNode?temp?=?queue.poll();
????????????????System.out.print(temp.val+"?");
????????????????if(temp.left?!=?null)??queue.add(temp.left);
????????????????if(temp.right?!=?null)?queue.add(temp.right);
????????????}
????????}
????}
}
参考链接:代码随想率