按层打印二叉树(去哪儿网2017春招真题)

题目描述:

给定一棵二叉树的前序(根、左、右)和中序(左、根、右)的打印结果,输出此二叉树按层(从左往右)打印结果。

例如一棵二叉树前序:1 2 4 5 3;中序:4 2 5 1 3。可以构建出下图所示二叉树:



按层打印的结果则为:1 2 3 4 5。

分析:1,根据先序遍历和中序遍历构建一棵新树。2,按层次遍历这个树

import java.util.*;
public class levelTraversal {
    TreeNode root ;
    TreeNode treeNode ;
//在中序遍历中找根节点
public int findPos(int[] inOrderT, int val){ for(int i=0;i<inOrderT.length;i++) if(val == inOrderT[i])return i;//在中序遍历中找出子树中的根节点 return 0; }
// 按层次打印
public void levelTraversal(TreeNode root){ if (root==null) { return; } Queue<TreeNode> queue = new LinkedList<TreeNode>(); queue.offer(root); TreeNode node ; while(queue.size()!=0){ node = queue.poll(); System.out.print(node.val); if(node.left!=null)queue.offer(node.left); if(node.right!=null)queue.offer(node.right); } }
//创建一棵新树
public TreeNode creatTree(int[] firstT, int fleft, int fright, int[] inOrderT, int ileft, int iright){ if(fleft>fright || ileft>iright) return null; TreeNode treeNode = new TreeNode(firstT[fleft]);//每次新建子树根节点作为父树的孩子节点 int i = findPos(inOrderT, firstT[fleft]);//找到中序遍历的根节点 treeNode.left = creatTree(firstT, fleft+1, fleft+i-ileft, inOrderT, ileft, i-1); treeNode.right = creatTree(firstT, fleft+i-ileft+1, fright, inOrderT, i+1, iright); return treeNode; } public static void main (String[] args) { Scanner sc = new Scanner(System.in); int n = sc.nextInt(); int[] a = new int[n]; int[] b = new int[n]; for(int i = 0;i<n;i++){ a[i] = sc.nextInt(); } for(int i = 0;i<n;i++){ b[i] = sc.nextInt(); } levelTraversal lt = new levelTraversal(); //TreeNode root=lt.newRoot(a); TreeNode root=lt.creatTree(a,0,a.length-1,b,0,b.length-1); lt.levelTraversal(root); } }
//一个树节点
class TreeNode{ int val; TreeNode left=null; TreeNode right=null; public TreeNode(int val){ this.val = val; } }

 

posted @ 2017-04-26 12:47  落叶已归根  阅读(374)  评论(0编辑  收藏  举报